452刪除列表中的元素

引用塊內容

特殊情況考慮

1.首節點爲空

if(head==NULL)
{
return 0;
}

2.尾節點爲空

1->null 1
結果應爲 null

3.1->null->1

/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
/**
* @param head a ListNode
* @param val an integer
* @return a ListNode
*/
ListNode *removeElements(ListNode *head, int val) {
if(head==NULL)
{
return NULL;
}

    ListNode*m=head;
    while(m->next!=NULL)
    {
        if(m->next->val==val)
        {
            m->next=m->next->next;
        }
        else
        {
            m=m->next;
        }
    }
    if(head->val==val)
    {
        if(head->next==NULL)
        {
            return NULL;
        }
    }
return head;   
}        

};