[题解] Leetcode 203 移除链表元素

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

二、解法

最直接的思路就是遍历一遍整个链表，遇到 Node.val==val时就删除该结点。

一般来说，删除一个链表结点的方式是找到待删除结点的上一个结点，然后修改上一个结点的指针域。假设指针p指向待删除结点的上一个结点，则用 p->next=p->next->next 即可删除该结点。

这道题的示例3提示了我们有待删除结点就是头结点的情况，这种情况下我们使用虚拟头结点的技巧。

/**
 * 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) {
        if(head==nullptr) return head;
        ListNode virnode(0,head),*p=&virnode;
        while(p->next){
            if(p->next->val==val) p->next=p->next->next;
            else p=p->next;//这里写else是为了预防两个数据域为val的结点相连的情况
        }
        return virnode.next;
    }
};