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

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/remove-linked-list-elements
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

• 其实这道题的思路是很清晰的，主要做法有迭代法和递归法.
• 在做迭代的时候我们要考虑一些特殊情况（头节点的val等于val）该怎么办。
• 其实有两个办法
• 法一：就是在开头写当头节点的val等于val时就让头节点赋值为头节点的next
• 法二：就是建立一个虚拟头节点就可以了。
• 最后的递归就不用考虑特殊情况了，直接写就可以了。

法一（迭代法）

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */


struct ListNode* removeElements(struct ListNode* head, int val){
    if(!head)
    {
        return head;
    }
    struct ListNode* prehead=(struct ListNode*)malloc(sizeof(struct ListNode));
    //我使用了一个虚拟头节点。//但是最后记得把虚拟头节点处理一下。
    prehead->next=head;
    struct ListNode* fast=head,*slow=prehead;
    while(fast!=NULL)
    {
        if(fast->val==val)
        {
            fast=fast->next;
            slow->next=fast;
        }
        else 
        {
            fast=fast->next;
            slow=slow->next;
        }
    }
    return prehead->next;
}

法二（递归）

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */


struct ListNode* removeElements(struct ListNode* head, int val){
    if(!head)
    {
        return head;
    }
    if(head->val==val)
    {
        head=removeElements(head->next,val);
        return head;
    }
    else
    {
        head->next=removeElements(head->next,val);
        return head;
    }
}