「Leetcode」203.移除链表元素

题目描述

删除链表中等于给定值 val 的所有节点。

示例:
输入: 1->2->6->3->4->5->6, val = 6
输出: 1->2->3->4->5

使用语言：C语言
思路解析

• 给两个指针cur和prev，prev是在cur前面的指针
• 找到链表中节点val值和输入val值相等的节点，若第一个节点就是需要删除的则直接让头指针指向下一个节点head=cur->next，删除第一个节点free(cur)，再让cur=head即可
• 若找到的需要删除的节点非第一个节点，则让prev->next=cur->next，删除该节点free(cur)，然后让cur=prev->next即可

代码实现

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

typedef struct ListNode Node;//后面只需要使用Node*

struct ListNode* removeElements(struct ListNode* head, int val)
{  
    Node* cur=head;
    Node* prev=NULL;

    while(cur)
    {
        if(cur->val==val)
        {
            if(cur == head)//删除的节点是第一个节点
            {
                head=cur->next;
                free(cur);
                cur=head;
            }
            else//删除非第一个节点
            {
                prev->next=cur->next;
                free(cur);
                cur=prev->next;
            }
        }
        else
        {
            prev=cur;
            cur=cur->next;
        }
    }
    return head;
}

运行结果