codeforce 767C Garland (树形dp)

题意：给你一棵树，每个节点都有一个权值，之后你可以删除两条边把他变成3颗子树现在使得他的三颗的权值相等。

思路：显然他们的点权都是3的倍数才会得到3颗权值相等的树，那么我们就树形dp，自下而上的求出以某个点为根的子树的权值，如果他的权值等于sum/3 那么就让这颗子树的权值为0，接着dfs就好了

代码：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e6+10;
vector<int>V[maxn];
vector<int>ANS;
int Size[maxn] , cur , W[maxn];
void dfs(int u ,int fa)
{
    Size[u] = W[u];
    for(int i = 0 ; i < V[u].size() ; i ++)
    {
        int v = V[u][i];
        if(v == fa) continue;
        dfs(v,u);
        Size[u] += Size[v];
    }
    if(Size[u] == cur) {ANS.push_back(u),Size[u] = 0;}
}
int main()
{
    int n;
    scanf("%d",&n);
    int pos , val , root , sum = 0;
    for(int i = 1 ; i <= n ; i++)
    {
        scanf("%d%d",&pos, &val);
        if(pos == 0) {root = i;}
        V[i].emplace_back(pos);
        V[pos].emplace_back(i);
        W[i] = val;
        sum += val;
    }
    if(sum%3) {puts("-1");return 0;}
    cur = sum/3;
    dfs(root,0);
    if(ANS.size() <= 2) {puts("-1"); return 0;}
    cout<<ANS[0]<<" "<<ANS[1]<<endl;

}
/*
4
0 1
1 -1
2 1
3 -1
*/