删除链表中的元素

1、问题描述

      删除链表中等于给定值val的所有节点。给出链表 1->2->3->3->4->5->3, 和 val = 3, 你需要返回删除3之后的链表：1->2->4->5。

2、实现思路

      遍历链表，若值等于val，此结点删除，否则依次检验。

3、代码

/**
 * 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) {
        // Write your code here
       ListNode *p= new ListNode;
        p->next = head;
        head=p;
        while (head->next != NULL) {
            if (head->next->val == val) {
                head->next = head->next->next;
            } else {
                head = head->next;
            }
        } 
        return p->next;
    }
};

4、感想

      定义新节点指向head，若head->next的值为val，head覆盖head->next，否则head后移。注意若没有head=p，会runtime error，要保留head。