这个题目的思路基本上和3Sum是类似的。使用三个指针,一个指针p1指向要计算的指定区间子链表的前一个节点,方便删除节点,即它自己不会参与计算;另外两个指针p2和p3则用于计算一个区间内的综合,如果某个区间内总和为0,那么此时p3所指的正好是区间外的第一个节点,此时将p1.next=p3
即可。代码如下:
public ListNode removeZeroSumSublists(ListNode head) {
if (head == null || (head.next == null && head.val != 0)) {
return head;
}
//要先构造一个头结点,作为初始的p0,所以要计算的区间为p0后面的节点,也就是最初的头结点所在区间
ListNode newHead = new ListNode(0);
newHead.next = head;
ListNode p1 = newHead, p2 = p1.next, p3;
while (p1 != null && p1.next != null) {
p2 = p1.next; //指定区间的开始,为p1.next
p3 = p2.next; //指定区间的结束标志
int curResult = p2.val;
while (p3 != null && curResult != 0) {
curResult += p3.val;
p3 = p3.next;
}
//如果这个区间内总和为0,则让p1指向p3
if (curResult == 0) {
p1.next = p3;
} else { //否则p1向后走,计算下一个区间
p1 = p1.next;
}
}
//最后将头结点去掉,即返回新头结点的next即可
return newHead.next;
}