难度: 中等
题目描述
给你一个链表的头节点 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
.
思路
遍历时记录前缀和及其位置,当某个前缀和重复出现时,两次出现之间的节点(不包括第一次出现的那个节点)都可以删掉,由于head也可以被删掉,所以在head之前再加一个节点。
代码
/**
* 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) {
bool ok = false;
ListNode *header = new ListNode(0);
header->next = head;
while (!ok) {
int sum = 0;
unordered_map<int, ListNode*> sum2node;
ok = true;
for (auto *iter = header; iter != nullptr; iter = iter->next) {
sum += iter->val;
if (sum2node[sum] != nullptr) {
ok = false;
int tmpSum = sum;
auto left = sum2node[sum], right = iter->next;
for (auto iter = left->next; iter != right; iter = iter->next)
sum2node[tmpSum += iter->val] = nullptr;
left->next = right;
sum2node[sum] = left;
} else sum2node[sum] = iter;
}
}
return header->next;
}
};