LeetCode 203 - 移除链表元素

news2024/10/21 14:21:20

题目描述

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

解题思路

创建一个虚拟头节点dummyHead，并将其next指向给定的头节点head，这样可以避免处理头节点的特殊情况。使用指针cur来遍历链表，当cur的下一个节点不为空时，进行如下操作：
1.如果cur的下一个节点的值等于给定的数值val，则将其下一个节点（即要移除的节点）保存在临时指针tmp中，然后将cur的next指针指向下下个节点，同时删除tmp指向的节点，完成移除操作。
2.如果cur的下一个节点的值不等于给定的数值val，则将cur指针指向下一个节点，即保持链表的连续性。
3.最后，将head指向dummyHead的下一个节点，即新的头节点，然后删除dummyHead节点释放内存，最终返回新的头节点。

算法实现

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*dummyHead=new ListNode(0);
        dummyHead->next=head;
        ListNode*cur=dummyHead;
        while(cur->next!=NULL)
        {
            if(cur->next->val==val)
            {
                ListNode*tmp=cur->next;
                cur->next=cur->next->next;
                delete tmp;
            }
            else
            {
                cur=cur->next;
            }
        }
        head=dummyHead->next;
        delete dummyHead;
        return head;
    }
};