树上dfs + 思维

the number lamp ai, it is hanging on (and 0, if is there is no such lamp), 表示的是祖先的序号

链接：点击打开链接

题目中的坑：应该自下而上的判断子树的累加和是否满足条件。如果是自上而下的判断，一旦该树的子树有满足条件的，去掉子树后又满足条件这种情况是无法识别出来。只能一回溯，且实际上回溯的复杂度并不高，相当于遍历所有点两遍。

解析：dfs 判断累加和。如果一个子树满足条件，价值变为零，两棵及以上满足条件则成功（恰好两棵需要额外判断不是根节点）

代码：

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
int const maxn = 1000006;

vector<int>son[maxn];
int t[maxn],p,root;
int pa[maxn],ans[2];
int num = 0;
void dfs(int x)
{
    if(num == 2)return;
    int temp;
    for(int i = 0; i < son[x].size(); i ++)
    {
        temp = son[x][i];
        dfs(temp);
        t[x] += t[temp];
    }
    if(t[x] == p && x != root)
    {
        ans[num] = x;
        num ++;
        t[x] = 0;
        if(num == 2)return;
    }
}
int main()
{
   // freopen("in.txt","r",stdin);
    int n;
    int sum = 0;
    scanf("%d",&n);

    for(int i = 1; i <= n; i ++)
    {
        scanf("%d%I64d",&pa[i],&t[i]);
        sum += t[i];
        son[pa[i]].push_back(i);
        if(pa[i] == 0)root = i;
    }
    bool flag = false;
    if(sum % 3 != 0 || sum == 0)
    {
        flag = true;
    }
    else
    {
        p = sum / 3;
        dfs(root);
        if(num >= 2)
        {
            printf("%d %d\n",ans[0],ans[1]);
        }
        else
        {
            flag = true;
        }
    }
    if(flag == true)
        printf("-1\n");
    return 0;
}