LeetCode 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, 104] 内
• 1 <= Node.val <= 50
• 0 <= val <= 50

思路

设置虚拟头节点，使得原链表的所有节点就都可以按照统一的方式进行移除。时间复杂度为O(n)。

代码

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* myhead = new ListNode(0); // 定义虚拟头节点
        myhead->next=head; // 将虚拟头结点指向head，这样方便后面做删除操作
        ListNode* cur=myhead;
        while(cur->next!=nullptr){
            if(cur->next->val==val){
                ListNode *tmp=cur->next;
                cur->next=cur->next->next;
                delete tmp;
            }else{
                cur=cur->next;
            }
        }
        head=myhead->next;
        delete myhead;
        return head;
    }
};

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]:
        # 创建虚拟头部节点以简化删除过程
        myhead=ListNode(next=head)
        cur=myhead
        # 遍历列表并删除值为val的节点
        while cur.next!=None:
            if cur.next.val==val:
                cur.next=cur.next.next
            else :
                cur=cur.next
        return myhead.next

需要注意的地方

1.使用虚拟头节点可以使得操作链表的方式变得统一。