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

给你一个链表的头节点 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]
 

提示:

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/remove-zero-sum-consecutive-nodes-from-linked-list
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* removeZeroSumSublists(ListNode* head) {
  if (!head)return head;
        ListNode* dummy = new ListNode(0, NULL);
        map<int, ListNode*> my_map;
        my_map[0] = dummy;
        int cur = 0, temp_cur;
        ListNode* p = head, * q,*p_pre=dummy;
        //ListNode *del;
        while (p)
        {
            cur += p->val;
            //若累加前缀不同
            if (my_map.find(cur) == my_map.end())
            {
                p_pre->next = p;
                my_map[cur] = p;
            }
            //若有相同的前缀
            else
            {
                //start 开始删除的前一个结点
                //temp 用来删除map中相应的记录
                temp_cur = cur;
                ListNode* start = my_map[cur];
                q = start->next;//第一个要删除的结点
                start->next=p->next;//把要删除的从链表中移除
                while (q&&q != p)
                {
                    temp_cur += q->val;
                    my_map.erase(temp_cur);
                    //del = q;
                    q = q->next;
                    //delete(del);
                }
                //delete(p);
                p = start;
            }
            p_pre = p;
            p = p->next;
        }
        return dummy->next;
    }
};

Leetcode 上我使用delete函数会报错,但具体原因没找到。

但是使用学校的oj,带delete函数是可以通过的。

原文地址:https://www.cnblogs.com/lancelee98/p/13225652.html