Garland CodeForces - 767C (dfs)

  题意就是给你一棵树，然后要求你将这棵树三等分，即选2个节点，将这2个节点与其父节点的边切掉，不能选根节点。

  做法就是，选判断一下总点权值tmp能不被三整出，能的话则可能可以三等分，否则的话肯定不能三等分。然后还要注意一下可能将树三等分可以切的点的组合（2个点为一个组合）有多个只要一种即可 ，从root开始往下dfs，回溯的时候更新一下节点对应的子树的点权值，若满足sum[now]==tmp，说明该节点是可以切去的，切掉了之后sum【now】要清0，因为切掉之和子树now对其父节点的贡献就是0了。ps：这题千万能用cin读入。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=1e6+10;
vector<int>g[maxn];
ll sum[maxn];
int val[maxn],ans[5],tmp,num,root,n;
void dfs(int now)
{

    sum[now]=val[now];
    int sz=g[now].size();
    for(int i=0;i<sz;i++)
    {
        int v=g[now][i];
        dfs(v);
        sum[now]+=sum[v];
    }
    if(root!=now&&sum[now]==tmp&&num!=2)
    {
        ans[num++]=now;
        sum[now]=0;
    }
}
int main()
{

    scanf("%d",&n);
    for(int i=1;i<=n;i++) g[i].clear();
    tmp=num=0;
    for(int i=1;i<=n;i++)
    {
        int u,v=i;
        scanf("%d %d",&u,&(val[i]));
        tmp+=val[i];
        if(u==0)
        {
            root=i;
            continue;
        }
        g[u].push_back(i);
    }
    if(tmp%3!=0)
    {
        cout<<-1<<endl;
        return 0;
    }
    tmp/=3;
    dfs(root);
    if(num==2)
        cout<<ans[0]<<" "<<ans[1]<<endl;
    else
        cout<<-1<<endl;
    return 0;
}