Leetcode 1171. 从链表中删去总和值为零的连续节点

发布时间 2023-06-11 18:49:41作者: DL1024

题目:

给你一个链表的头节点 head,请你编写代码,反复删去链表中由 总和 值为 0 的连续节点组成的序列,直到不存在这样的序列为止。

删除完毕后,请你返回最终结果链表的头节点。

你可以返回任何满足题目要求的答案。

(注意,下面示例中的所有序列,都是对 ListNode 对象序列化的表示。)

难度:中等

示例1:

输入:head = [1,2,-3,3,1]
输出:[3,1]
提示:答案 [1,2,1] 也是正确的。

示例2:

输入:head = [1,2,3,-3,4]
输出:[1,2,4]

示例3:

输入:head = [1,2,3,-3,-2]
输出:[1]

提示:

  • 给你的链表中可能有 11000 个节点。
  • 对于链表中的每个节点,节点的值:-1000 <= node.val <= 1000.

代码实现:

方法一:递归

class Solution {
public:
    ListNode* removeZeroSumSublists(ListNode* head) {
        // 判空
        if(!head) return head;
        int sum = 0;
        // 每一个节点 都进行判断是否存在该节点开始的 连续节点合为 0
        for(ListNode* p = head; p != nullptr; p = p->next){
            if((sum += p->val) == 0){
                return removeZeroSumSublists(p->next);
            }
        }
        head->next = removeZeroSumSublists(head->next);
        return head;
    }
};

方法二:哈希

class Solution {
public:
    ListNode* removeZeroSumSublists(ListNode* head) {

        unordered_map<int, ListNode*> mp;   // 存放 前缀和, 以及该节点指针

        ListNode* root = new ListNode(0, head); // 辅助头结点 防止整个链表合为0
        int sum = 0;
        ListNode* p = root;
        for(; p; p = p->next){
            sum += p->val;
            mp[sum] = p;
        }

        for(p = root, sum = 0; p; p = p->next){
            sum += p->val;
            if(mp.count(sum)){  // 存在其他节点和 与 当前相同,则直接跳过中间这一段,往后找
                p->next = mp[sum]->next;
            }
        }
        return root->next;
    }
};