【LeetCode】28.Linked List— Remove Linked List Elements删除链表元素

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
        ListNode*p=head;      //定义前驱结点指针
        if(p==NULL) return NULL;
        ListNode*q=p->next;   //定义要删除的结点指针
        while(1)
        {
            if(q==NULL){ //如果链表只有一个结点且头结点是要删除的结点
                if(head->val==val)
                {
                    delete p;
                    return NULL;
                }
                else return head;
            }
            else{      //如果链表不止一个结点
                if(head->val==val)  //头结点是要删除的结点
                {
                    head=q;
                    delete p;
                    p=head;
                    q=p->next;
                }
                else if(q->val==val)  //找到要删除的结点
                {
                    p->next=q->next;
                    delete q;
                    q=p->next;           
                }
                else{          //没有要删除的结点，则一头后移。
                    p=p->next;
                    q=p->next;
                }
            }
        }
        return head;
    }
};