题目:
给你一个链表的头节点 head,请你编写代码,反复删去链表中由 总和 值为 0 的连续节点组成的序列,直到不存在这样的序列为止。
删除完毕后,请你返回最终结果链表的头节点。
示例:
输入:head = [1,2,-3,3,1] 输出:[3,1] 提示:答案 [1,2,1] 也是正确的。
代码如下:
class Solution {
public ListNode removeZeroSumSublists(ListNode head) {
if(head==null) return null;
ListNode prev=new ListNode(0);
prev.next=head;
Map<Integer,ListNode>map=new HashMap<>();//存放截至到该节点的节点和
int sum=0;
for(ListNode p=prev;p!=null;p=p.next){
sum+=p.val;
map.put(sum,p);//会覆盖掉之前和为sum的节点
}
sum=0;
for(ListNode q=prev;q!=null;q=q.next){//这里从prev开始遍历,排除类似于[1,-1]的情况
sum+=q.val;
q.next=map.get(sum).next;//跳过中间和为0的节点
}
return prev.next;
}
}