CodeForces - 767C Garland（附带易错数据）

题目链接：Garland

题意：

给一颗树，每个点有点权（可为负），删除两个边后的三个树的点权和相等。

思路：

这题要是看代码直接码的话，就跟没做一样，很多细节必须了解清楚才算做过。

首先sum%3！=0输出-1

是3的倍数的话，从下往上回溯，有一个子树和为sum/3，这个子树的顶点一定可以为第一个答案，大概说下原因

第一个树不会比这个树少，因为达不到sum/3。如果这个树还有一部分，那么和一定是为0的，那么这部分的0给其他树也行。

这个树为第一个树后，往上回溯的该子树和也标记为0，直到再==sum/3，这时候基本就是有答案了，因为剩下的一定联通且和一定是sum/3,只是剩余的一定要至少有一个点

易错数据

4
0 -3
1 2
2 1
2 0

输出

-1

这只有两个树和为0

那么他要进入（子树和==sum/3）三次才行，后面的就算还有和也一定是0。

代码

#include<bits/stdc++.h>
using namespace std;
#define fo(a,b) for(int i=a;i<=b;i++)
#define ll long long
#define dou double
#define EXP 1e-8
#define M 1000005
int n,sum=0;
int dp[M];
int f[4],s=0;
struct Node
{
    vector<pair<int,int>>v;
}x[M];
void dfs(int d,vector<pair<int,int>>&v,int pre,int data)
{
    dp[d]=data;
    int l=v.size();
    for(int i=0;i<l;i++){
        int now=v[i].first;
        int data=v[i].second;
        if(now!=pre){
            dfs(now,x[now].v,d,data);
            dp[d]+=dp[now];
        }
    }
    //等子树和加完再判断
    if(s==4) return;
    if(dp[d]==sum/3&&f[s]==0){        //要进入三次
        f[s++]=d;
        dp[d]=0;
    }
}
signed main()
{
    cin>>n;
    for(int a,b,i=1;i<=n;i++){
        scanf("%d%d",&a,&b);
        x[i].v.push_back(make_pair(a,b));
        x[a].v.push_back(make_pair(i,b));
        sum+=b;
    }
    if(sum%3==0){
        dfs(0,x[0].v,-1,0);
        if(f[0]&&f[1]&&f[2]) {cout<<f[0]<<" "<<f[1]<<endl;return 0;}
    }
    cout<<-1<<endl;
    return 0;
}