( 链表) 203. 移除链表元素 ——【Leetcode每日一题】

news2024/11/25 6:37:59

❓203. 移除链表元素

难度：简单

给你一个链表的头节点 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
输出：[]

提示：

列表中的节点数目在范围 $0, 10^4]$ 内
1 <= Node.val <= 50
0 <= val <= 50

💡思路：

添加一个虚拟头结点，可以统一化处理！

🍁代码：(Java、C++)

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 h = new ListNode();
        h.next = head;
        head = h;
        while(h.next != null){
            if(h.next.val == val){
                h.next = h.next.next;
            }else{
                h = h.next;
            }
        }
        return head.next;
    }
}

C++

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
        ListNode* h = new ListNode();
        h->next = head;
        head = h;
        while(h->next != nullptr){
            if(h->next->val == val){
                h->next = h->next->next;
            }else{
                h = h->next;
            }
        }
        h = head->next;
        delete head;
        return h;
    }
};