Codeforces 767C Garland （dfs）

题意：有一棵树，每个节点都有一个权值，需要分成相等的三部分，能则输出剪下的结点否则输出-1（有一个坑是不能剪下根节点）

题解：算是看了大佬的思路。首先判断是否sum为3的倍数。      dfs子节点的权值累加，如果找到一个sum/3，记录该结点然后将这个结点清0，因为有一种情况是某一个父节点和子节点都可能是sum/3，否则会出错。 找到2个则输出否则输出-1。  我看着网上是dfs两次的，然后我觉得dfs一次就可以，试了试果然AC了

代码：

#include <set>
#include <cmath>
#include <queue>
#include <stack>
#include <vector>
#include <string>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
const int maxn = 1000000 + 5;
int n;
ll sum;
int root;
vector <int> v[maxn];
int tmp[maxn];
int flag1 = 0;
int flag2 = 0;
void solve(int x)
{
    for(int i=0; i<v[x].size(); i++)
    {
        if(flag1 && flag2)
            return;
        solve(v[x][i]);
        tmp[x] += tmp[v[x][i]];
    }
     if(flag1 == 0 && tmp[x] == sum/3 && root!=x)
        {
            flag1 = x;
            tmp[x] = 0;
        }
    else if(flag2 == 0 && tmp[x] == sum/3 && root!=x)
        {
            flag2 = x;
            tmp[x] = 0;
        }
        return ;
}
int main()
{
    while(scanf("%d",&n)!=EOF && n)
    {
        sum = 0;
        memset(tmp,0,sizeof(tmp));
         root = 1;
        for(int i=1; i<=n; i++)
        {
            int a,t;
            scanf("%d%d",&a,&t);
            sum += t;
            if(a!=0)
            v[a].push_back(i);
            else root = i;
            tmp[i] = t;
        }
        if(sum%3 != 0)
        {
            printf("-1\n");
            continue;
        }
        flag1 = flag2 = 0;
        solve(root);
        if(flag1 && flag2)
            printf("%d %d\n",flag1,flag2);
        else
            printf("-1\n");

    }
}