乐扣T1171 OJ9：从链表中删去总和值为0的连续节点 c++

问题描述 :

给你一个链表的头节点 head，请你编写代码，反复删去链表中由总和值为 0 的连续节点组成的序列，直到不存在这样的序列为止。如果存在多个总和值为0的连续节点序列，优先删除位置靠前的序列。

删除完毕后，请你返回最终结果链表的头节点。

示例 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.

输入范例：
5
1 2 -3 3 1

输出范例：
head–>3–>1–>tail

#include<iostream>
using namespace std;
struct ListNode{
    int val;
    struct ListNode *next;
    ListNode() : val(0), next(NULL) {}
    ListNode(int x) : val(x), next(NULL) {}
    ListNode(int x, ListNode *next) : val(x), next(next) {}
};
class Solution
{
public:
    ListNode* removeZeroSumSublists(ListNode* head) {
             //填充本函数完成功能
        ListNode *dummy=new ListNode(-1);
        dummy->next=head;
        ListNode *p=dummy;
        ListNode *q=p->next;

        while(p->next){
            int sum=0;
            while(q){
                sum=sum+q->val;
                if(sum==0){
                    break;
                }
                q=q->next;
            }
            if(sum==0){
                q=q->next;
                p->next=q;
            }
            else{
                p=p->next;
                q=p->next;
            }
        }
        return dummy->next;
    }
};
ListNode *createByTail()
{
    ListNode *head;
    ListNode *p1,*p2;
    int n=0,num;
    int len;
    cin>>len;
    head=NULL;
    while(n<len )
    {
        cin>>num;
        p1=new ListNode(num);
        n=n+1;
        if(n==1)
            head=p1;
        else
            p2->next=p1;
        p2=p1;
    }
    return head;
}
void  displayLink(ListNode *head)
{
    ListNode *p;
    p=head;
    cout<<"head-->";
    while(p!= NULL)
    {
        cout<<p->val<<"-->";
        p=p->next;
    }
    cout<<"tail\n";
}
int main()
{
    ListNode* head = createByTail();
    head=Solution().removeZeroSumSublists(head);
    displayLink(head);
    return 0;
}