力扣203.移除链表元素

大白飞飞2024-10-07 12:44

题目链接：203. 移除链表元素 - 力扣（LeetCode）

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

示例 1：

输入： 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
输出：[]

用原来的链表操作：

要删除的是curr.next

java 复制代码

/**
 * 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) {
        //要删的是头结点
        while(head!=null&&head.val==val){ //用while不用if eg:1,1,1,1,1
            head=head.next;
        }
        //要删的不是头结点
        ListNode curr=head; //定义一个临时指针指向head，使用curr进行遍历，避免修改head
        while(curr!=null&&curr.next!=null){
            if(curr.next.val==val){
                curr.next=curr.next.next;
            }else{
                curr=curr.next;
            }
        }
        return head; //没有改变head
    }
}

设置一个虚拟头结点：

要删除的是curr.next

java 复制代码

/**
 * 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();
        dummyhead.next=head; //虚拟头结点的下一个节点指向头结点head
        ListNode curr=dummyhead; //定义一个临时指针指向虚拟头结点
        while(curr!=null&&curr.next!=null){
            if(curr.next.val==val){
                curr.next=curr.next.next;
            }else{
                curr=curr.next;
            }
        }
        return dummyhead.next; // 头结点head可能已经被删除了，dummyhead.next是新链表的头结点
    }
}

注意不能直接用虚拟头结点dummyhead遍历。

因为循环条件while(dummyhead!=null&&dummyhead.next!=null)在处理链表的最后一个节点时可能会出现问题。当dummyhead.next为null时，循环会停止，但是如果最后一个节点的值等于val，这个节点将不会被删除。