LeetCode Remove Linked List Elements 链表-优快云博客

本文链接：https://blog.youkuaiyun.com/yeruby/article/details/49309841

本文介绍使用C++进行链表操作时，如何优化删除元素的代码，避免内存泄露，提供了一种更简洁高效的实现方式，并附带了讨论区中更优解的对比分析。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

思路：

注意用c++进行链表的删除时，对相应node需要free，避免内存泄露。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
        if(head == NULL) return head;
        ListNode yummy(-1);
        ListNode *p = &yummy, *q = head;
        while(q != NULL) {
            if(q->val != val) {//not equal to link
                p->next = q;
                p = q;
                q = q->next;
            }else {//equal not to link
                auto freeNode = q;
                q = q->next;
                free(freeNode);
                if(p == &yummy) {
                    head = q;
                }
            }
        }
        p->next = q;
        return head;
    }
};

DISCUSS里的一段更漂亮的code，自愧不如：

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
        if (head==NULL){
            return head;
        }
        ListNode* p=head;
        while(p->next!=NULL){
            if (p->next->val == val){
                auto freeNode = p->next;
                p->next = p->next->next;
                free(freeNode);
            }
            else{
                p = p->next;
            }
        }
        //下面的code非常巧妙配合上面的if分支
        if (head->val == val)
            head = head->next;
        return head;
    }
};