代码随想录17--移除链表元素（力扣 203）--刷题笔记

移除链表元素（力扣 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

代码：

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

/*
    思路：
        使用虚拟头结点
*/

struct ListNode* removeElements(struct ListNode* head, int val){
    struct ListNode *prior = NULL;  /*指向当前节点的前一个节点*/
    struct ListNode *cur = NULL;    /*指向当前节点*/
    /*创建虚拟头节点*/
    struct ListNode * hNode = malloc(sizeof(struct ListNode ));
    hNode->next = head;
    prior = hNode;
    while(prior->next){
        cur = prior->next;
        if(cur->val == val){ /*需要删除cur*/
            prior->next = cur->next;  /*将前一个节点的指针指向后一个节点*/
            free(cur); /*释放删除指针的资源，且删除后不需要移动prior*/
        }else{
            prior = prior->next;  /*不需要删除则继续遍历*/
        }
         

    }
    /*不要忘了去虚拟头结点*/
    head = hNode->next;
    free(hNode);
    return head;

}