给你一个链表的头节点 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]
/**
* 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 nullptr;
//如果当前元素为零, 直接返回其后面链表经过函数处理后的结果,相当于删除当前节点
if(!head->val) removeZeroSumSublists(head->next);
//如果当前元素不为0,直接对其后面的链表进行护理,主义,处理过后,链表不包含任何和为0 的连续序列
head->next = removeZeroSumSublists(head->next);
//引入头结点
ListNode dummy(0);
dummy.next = head;
//由于从第二个元素开始后面不包含任何和为0的连续序列,如果出现连续序列和为0, 一定是第一个元素和后面某个连续序列的和
for(int sum = 0; head; head= head->next){
sum += head->val;
//从第一个元素开始求和,一旦和为0,那么从链表开头到当前元素这段链表就可以删除了
if(!sum) dummy.next= head->next;
}
return dummy.next;
}
};
提示:
- 给你的链表中可能有
1
到1000
个节点。 - 对于链表中的每个节点,节点的值:
-1000 <= node.val <= 1000
.