目录链接:
力扣编程题-解法汇总_分享+记录-CSDN博客
GitHub同步刷题项目:
https://github.com/September26/java-algorithms
原题链接:力扣
描述:
给你一个链表的头节点 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
.
解题思路:
* 解题思路:
* 构建前缀和,相同前缀和的位置,其连续累加值必定为0,相同前缀和之间的节点都是可以被删掉的。
* 另外,如果有多个前缀和都相同的节点,我们只需要取第一个和最后一个即可。
* 所以构建一个map,key为前缀和,value则为对应的节点。
* 首先遍历第一遍,找到最后一个前缀和的节点加入到map中。
* 然后第二遍循环,找到等于当前前缀和值的节点,这时候是一定存在的,是不过有可能是当前的节点。
* 如果是当前节点的话,使用node->next = seen[prefix]->next;也满足要求。
代码:
class Solution1171
{
public:
ListNode *removeZeroSumSublists(ListNode *head)
{
ListNode *newHead = new ListNode(0);
newHead->next = head;
int prefix = 0;
map<int, ListNode *> seen;
for (ListNode *node = newHead; node; node = node->next)
{
prefix += node->val;
seen[prefix] = node;
}
prefix = 0;
for (ListNode *node = newHead; node; node = node->next)
{
prefix += node->val;
node->next = seen[prefix]->next;
}
return newHead->next;
}
};