题意就是给你一棵树,然后要求你将这棵树三等分,即选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;
}