代码随想录算法训练营打卡day03-1|203 移除链表元素（虚拟头节点）

学习来自《代码随想录》

建议： 本题最关键是要理解 虚拟头结点的使用技巧，这个对链表题目很重要。

题目链接/文章讲解/视频讲解：：代码随想录

题目：力扣题目链接

视频：链表基础操作| LeetCode：203.移除链表元素

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

1 题目描述 

给你一个链表的头节点 head 和一个整数 val ，请你删除链表中所有满足 Node.val == val 的节点，并返回 新的头节点 。

输入：head = [1,2,6,3,4,5,6], val = 6

输出：[1,2,3,4,5]

示例 2：

输入：head = [], val = 1

输出：[]

示例 3：

输入：head = [7,7,7,7], val = 7

输出：[]

2 题解

方法一：分节点是头节点还是非头节点情况删除满足条件的节点

思路

在原来的链表上进行查询节点并删除值=val的节点。

此时需要考虑两种情况的节点

1、要删的是头节点

删除步骤：原头节点head 后移一个节点即可（head =head.next），原头节点会被系统gc机制回收掉，不要处理（c++需要手动回收）。

注意：head要移到第一个值不为val的节点才算删除完，而不是删一个就行了。

例如[1,1,1,1,3,4,5,2,1,6] val=1 删头节点最后head要指向3，所以需要用循环语句实现

2、要删是 非头节点

删除步骤：要删除节点的上一个节点改指向要删除节点的下一个节点，头节点不变。需要依赖前一个节点来操作。

代码

java版本

注意1：此时注意到代码while(cur != null && cur.next != null){...}的判断条件中，cur遍历不到尾节点，就会问尾节点=val时怎么办能删除吗？实际上再cur指向的是目标节点的上一个节点，即cur为倒数第二个节点时，就能处理尾节点=cal的情况了。

注意2：注意两个while循环的空节点的判断条件，否则会有空指针的情况。

/**
 * 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) {
      
        //循环删除链表头为val的节点，直到指向第一个不为val的节点
        //因为语句中需要取head.val，因此head不能为空节点，否则空指针
        while(head != null && head.val == val){
            head = head.next;
        }
        
        ListNode cur = head;
        //因为需要取cur.next，因此cur不能为空节点
        //因为需要取cur.next.next，因此cur.next不能为空节点。
        while(cur != null && cur.next != null){
            if(cur.next.val == val){
                cur.next = cur.next.next;
            }else{
                cur = cur.next;
            }
        }
        return head;
    }
}

复杂度分析：

时间复杂度：O(n) 遍历长度为n的链表

空间复杂度： O(1)

python 版本：

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def removeElements(self, head: Optional[ListNode], val: int) -> Optional[ListNode]:
        while head != None and head.val == val:
            head = head.next
        cur = ListNode(next = head)
        while cur != None and cur.next != None:
            if cur.next.val == val:
                cur.next = cur.next.next
            else:
                cur = cur.next
        return head
            

方法二：设置虚拟头节点（推荐）

思路

在方法一的基础上进行改进，方法一种因为头节点和非头节点的操作不同，是因为头节点没有上一个节点，无法统一逻辑。那我们给头节点人为添加一个虚拟的头节点，原头节点就成了普通节点，就能解决逻辑不一致的问题，进而简化的代码。

代码

java版本

需要注意的点：

1、 虚拟头节点dummyHead 是加在原头节点前的。删除后的新链表的头节点是dummyHead.next，原来的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) {
        ListNode dummyHead = new ListNode(0);//设置虚拟头节点，值随便设置
        dummyHead.next = head;//虚拟节点指向头结点，这样头节点变成了中间节点，和其他中间节点一样操作，不用单独考虑
        ListNode cur = new ListNode();//临时节点
        cur = dummyHead;
        //因为下面会有cur.next.next所以这里cur.next不可以为null，cur最多指向倒数第2个节点
        while(cur.next != null){
            if(cur.next.val == val){
                //如果下个节点值等于val，则将节点指向下下个节点，剔除下个节点
                cur.next = cur.next.next;
            }else{
                //如果下个节点值不等于val,则判断下一个节点。删除的节点会被GC掉
                cur = cur.next;
            }
        }
        //重新设置头节点，并返回，也可以直接返回dummyHead.next
        head = dummyHead.next;
        return head;
    }

复杂度分析：

时间复杂度：O(n) 遍历长度为n的链表

空间复杂度： O(1)

python 版本：

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def removeElements(self, head: Optional[ListNode], val: int) -> Optional[ListNode]:
        dummyHead = ListNode(next=head)#定义一个虚拟头节点，加在旧头节点前面
        cur = dummyHead #临时节点，用于遍历链表

        while cur != None and cur.next != None:
            if cur.next.val == val:
                cur.next = cur.next.next
            else:
                cur = cur.next
        return dummyHead.next

3 总结

把特殊情况非特殊化，简化考虑情况。

链表要注意判空条件，否则容易出现空指针。