【数据结构】链表相关练习题:移除链表元素

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

  • 示例:

输入: 1->2->6->3->4->5->6, val = 6
输出: 1->2->3->4->5
  • 分析:

删除链表中给定的某个数,首先能想到的是把链表遍历一遍,找一个删一个,但是有一个问题,链表呢有一个特点,就是它前一个节点会保存指向下一个节点的指针,如果我们直接删掉,会使得前面的节点找不到下一个节点,所以说我们得定义一个指针用来保存前一个节点,然后再删除指定的节点。

  • 具体代码如下:

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */
struct ListNode* removeElements(struct ListNode* head, int val) {
    if(head==NULL)
        return head;
    struct ListNode* cur=head;
    struct ListNode* prev=NULL;
    
    while(cur)
    {
        if(cur->val!=val)
        {
            prev=cur;
            cur=cur->next;
        }
        else
        {
            struct ListNode* next=cur->next;
            if(prev==NULL)
            {
                head=cur->next;
            }
            else
            {
                 prev->next=cur->next; 
            }
            free(cur);
            cur=next;
        }
       
    }
    return head;
}

想要用oj写代码的同学可以点这个链接哦(快去测测你的代码能力啦):

https://leetcode-cn.com/problems/remove-linked-list-elements/description/

猜你喜欢

转载自blog.csdn.net/qq_42270373/article/details/83034127