Leetcode 203：Remove Linked List Elements

Remove all elements from a linked list of integers that have value val.

要点：

• 考虑如果 head.val == val 要怎么办？

[法1] 虚拟头结点

思路

题目非常简单，就是遍历链表，然后判断 node.val 是否等于 val，如果相等，就删除掉 node，如果不等，就继续遍历。

唯一需要注意的点是如果 head.val == val 就成立了，那么就需要更换头结点了。

我们当然可以单独处理这种情况，但是如果 head.val == val 成立后，我们更换了头结点，而新的头结点也满足这个条件的话，就又需要处理了。所以会比较繁琐。

比较好的思路是建立一个虚拟的头结点 dummyHead，就可以兼容所有的情况了，不需要对 head 做特殊处理。

代码

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode() {}
 *     ListNode(int val) { this.val = val; }
 *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
class Solution {

    public ListNode removeElements(ListNode head, int val) {
        
        if(head == null){
            return head;
        }

        //虚拟头结点
        ListNode dummyHead = new ListNode(0,head);
        //前一个结点
        ListNode pre = dummyHead;
        //当前结点
        ListNode cur = head;

        while(cur != null){
            //先保存下一个节点
            ListNode next = cur.next;
            //不相等，继续往后
            if(cur.val != val){
                pre = cur;
                cur = next;
            }else{
                //相等，删除当前元素
                pre.next = next;
                cur = next;
            }
        }

        return dummyHead.next;

    }
}

提交结果

代码分析

• 时间复杂度：O(n)
• 空间复杂度：O(1)