力扣 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;
}
};
csuzhucong 发布了1288 篇原创文章 · 获赞 689 · 访问量 173万+ 关注
标签:head,OJ,1171,示例,list,next,链表,节点 来源: https://blog.csdn.net/nameofcsdn/article/details/104605777