题意:有一棵树,每个节点都有一个权值,需要分成相等的三部分,能则输出剪下的结点否则输出-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");
}
}