【LeetCode】每日一题:移除链表元素 [C语言实现]

【LeetCode】每日一题:移除链表元素 [C语言实现]_第1张图片

内容专栏:《LeetCode刷题专栏》

本文概括:203.移除链表元素

本文作者:花 碟

发布时间:2023.5.5

 203. 移除链表元素 - 力扣(LeetCode)

题目:

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

示例1:

 【LeetCode】每日一题:移除链表元素 [C语言实现]_第2张图片

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

这种题型其实就是间接考察大家对于单链表的增删改查的操作,如果对于链表足够熟悉,这种题目做起来就不难理解,做起来就可以得心应手啦~

下面给出两种解题思路:

1.直接使用原来的链表来进行删除操作。(直接删除)

2.设置一个虚拟头结点再进行删除操作。(创建虚拟头节点)

思想1:直接删除

解释:直接删除就是在原来头节点的位置之上,直接进行删除等于val的节点,我们可以给予两个指针,cur指针用来删除等于val的节点,prev指针用来使删除cur的上一节点指向cur的下一节点,然后free掉cur,但是还有一种例外,如果第一个是等于val,我们想要删除的元素呢?所以我们在遍历整个链表之中,需要对以上两种情况进行考虑。

【LeetCode】每日一题:移除链表元素 [C语言实现]_第3张图片

 代码实现:

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */
struct ListNode* removeElements(struct ListNode* head, int val){
     struct ListNode* cur = head,* prev = NULL;
     while(cur != NULL)
     {
         //删除
         if(cur->val == val)
         {
             if(prev == NULL)
             { //1.头节点的值等于val
                cur = head->next;
                free(head); //删除头节点
                head = cur;
             }else
             {
                 //2.删除等于val的节点
                 prev->next = cur->next;
                 free(cur);
                 cur = prev->next;
             }
         }else
         {
             //往后继续寻找
             prev = cur;
             cur = cur->next;
         }
     }
     return head;
}

思想2:创建虚拟头节点

解释:我们可以创建一个新(虚拟)的头节点,利用我们学过的单链表尾插的形式,将原头节点head开始不等于val的节点进行尾插(分为空链表和非空链表两种情况),不等于val的节点在往后寻找前进行free掉即可。我们利用cur指针遍历原链表,用tail指针记录每次尾插后的位置,以免下次尾插需要回头再来寻找,降低了时间复杂度。最重要的一点是,在cur往后挪动之后,我们还需要将虚拟头节点的tail->next置为空,否则会出现野指针的情况。

【LeetCode】每日一题:移除链表元素 [C语言实现]_第4张图片

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */
struct ListNode* removeElements(struct ListNode* head, int val){
     struct ListNode* cur = head;
     struct ListNode* newhead = NULL,* tail = NULL;
     while(cur != NULL)
     {
          if(cur->val != val)
          {
              //头插(空链表和非空链表)
              if(tail == NULL)
              {
                  newhead = tail = cur;
              }
              else
              {
                  tail->next = cur;
                  tail = tail->next;
              }
              //cur往后走,tail的next一定得置空,因为尾节点必须置空
              //(可以将code1在循环结束code2的位置置空)。
              cur = cur->next;
              tail->next = NULL; //code1
          }
          else
          {
              //删除掉等于val的节点,往后寻找
              struct ListNode* del  = cur;
              cur = cur->next;
              free(del);
          }
     }
     
     /* // code2
        if(tail)
        tail->next = NULL;
     */
    return newhead;
}

你可能感兴趣的:(LeetCode刷题,链表,leetcode,数据结构)