Leetcode刷题记录 203. 移除链表元素

最新推荐文章于 2024-11-15 16:27:37 发布

dadadalolida

最新推荐文章于 2024-11-15 16:27:37 发布

阅读量421

点赞数

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.csdn.net/qq_48257950/article/details/127549777

版权

leetcode 专栏收录该内容

9 篇文章 0 订阅

订阅专栏

移除链表元素
给你一个链表的头节点 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

自己的做法：通过构建一个新链表来实现链表元素的移除。

struct ListNode* removeElements(struct ListNode* head, int val){
    struct ListNode* pre;
    struct ListNode* newlist;
    struct ListNode* node;
    //申请空间之后，一定要给指针赋值，否则会报错
    pre=(struct ListNode*)malloc(sizeof(struct ListNode));  //pre为头指针
    pre->next=NULL;
    newlist=pre;
    if(head==NULL){
        return head;
    }
    while(head){
        if(head->val!=val){
            node=(struct ListNode*)malloc(sizeof(struct ListNode));
            node->val=head->val;
            //申请空间之后，一定要给指针赋值，否则会报错
            node->next=NULL;
            pre->next=node;
            pre=node;
        }
        head=head->next;
    }
    return newlist->next;
}

看到讨论区比较有启发的题解，通过增加一个头结点的方式，方便直接在原链表上进行元素的移除。

struct ListNode* removeElements(struct ListNode* head, int val){
    struct ListNode * pHead = (struct ListNode *)malloc(sizeof(struct ListNode));
    pHead->next = head; //此处为增加一个头结点的代码

    struct ListNode * pWork = pHead;

    while (pWork!= NULL && pWork->next != NULL) {
        if (pWork->next->val == val) {
            pWork->next = pWork->next->next;
            continue;
        }
        pWork = pWork->next;
    }

    return pHead->next;
}