203. 移除链表元素 （虚拟头结点）

题目

203. 移除链表元素
     删除链表中等于给定值 val 的所有节点。

示例:

输入: 1->2->6->3->4->5->6, val = 6
输出: 1->2->3->4->5

常规解法

要判断删除的是否为头结点。

class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
   		 //头结点需要删除
        while (head != NULL && head -> val == val) { //注意！这里不是if！
            ListNode* tmp = head;
            head = head -> next;
            delete tmp;
        }
        //非头结点需要删除
        ListNode* p = head;
        while (p != NULL && p -> next != NULL) {
            if (p -> next -> val == val) {
                ListNode* tmp = p -> next;
                p -> next = p -> next -> next;
                delete tmp;
            } else {
                p = p -> next;
            }
        }
        return head;

    }
};

虚拟头结点

设置一个虚拟头结点，其next指向原来的头结点。这样就不用判断头结点删除的情况了。
注意：最终返回的是虚拟头结点的next。

class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
        ListNode* phead = new ListNode(0);
        phead -> next = head;
        ListNode* cur = phead;
        while (cur != NULL && cur -> next != NULL) {
            if (cur -> next -> val == val) {
                ListNode* tmp = cur -> next;
                cur -> next = cur -> next -> next;
                delete tmp;
            } else {
                cur = cur -> next;
            }
        }
        return phead -> next;
    }
};