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

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

问题描述 :

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

输入说明 :
首先输入链表长度len，然后输入len个整数，以空格分隔。

输出说明 :
输出格式见范例

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

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

#include<iostream>
#include<vector>
using namespace std;
struct ListNode
{
    int val;
    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:
    void countZeroSum(ListNode* node, int& sum, ListNode*& p2){
    	if(node==NULL) return;
    	sum+=node->val;
    	if(sum==0)
    		p2=node;//标记断点
    	countZeroSum(node->next,sum,p2);
    }
    ListNode* removeZeroSumSublists(ListNode* head) {
    	if(head==NULL) return head;
       ListNode *newhead=new ListNode(0,head);
       ListNode *p1;
       p1=newhead;
       while(p1!=NULL)
       {
       	int sum=0;
       	ListNode *p2;
       	p2=NULL;
       	countZeroSum(p1->next,sum,p2);
       	if(p2!=NULL)
       	{
       		p1->next=p2->next;
		}
		p1=p1->next;
	   }
	   return newhead->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;

}

说明：

给你的链表中可能有 1 到 1000 个节点。

对于链表中的每个节点，节点的值：-1000 <= node.val <= 1000.