力扣 OJ 1171. 从链表中删去总和值为零的连续节点

题目：

给你一个链表的头节点 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的一段。

代码：

class Solution {
public:
	ListNode* removeZeroSumSublists(ListNode* head) {
		int s[1005] = { 0 }, k = 0; //s是链表前缀和，k是链表长度
		ListNode*p = head, *list[1005];
		while (p)
		{
			list[++k] = p, s[k] = p->val + s[k - 1], p = p->next;
		}
		for (int i = 0; i < k; i++)
		{
			for (int j = k; j > i; j--)
			{
				if (s[i] != s[j])continue;
				if (i == 0)head = list[j]->next;
				else list[i]->next = list[j]->next;
				i = j;
			}
		}
		return head;
	}
};