题意:给你一棵树,每个节点都有一个权值,之后你可以删除两条边把他变成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
*/