leetcode 1171. 从链表中删去总和值为零的连续节点（C++）

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

C++

/**
 * 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) 
    {
        vector<int> tmp;
        ListNode* p=head;
        while(p)
        {
            tmp.push_back(p->val);
            p=p->next;
        }
        vector<int> res;
        for(auto it:tmp)
        {
            res.push_back(it);
            int sum=0;
            int m=res.size();
            for(int k=m-1;k>=0;k--)
            {
                sum+=res[k];
                if(0==sum)
                {
                    int count=m-k;
                    while(count)
                    {
                        res.pop_back();
                        count--;
                    }
                    break;
                }
            }                
        }
        ListNode* pnew=new ListNode(-1);
        ListNode* q=pnew;
        for(auto it:res)
        {
            q->next=new ListNode(it);
            q=q->next;
        }
        return pnew->next;
    }
};