Description
给出一棵 n n 个节点的有向树,每个点有点权,要求减掉两个非根节点与其父亲节点之间的边把树分成三部分使得这三部分非空且点权和相同
Input
第一行一整数表示点数,之后 n n 行每行输入节点的父亲节点 fa[i] f a [ i ] 和其点权 v[i] v [ i ]
(3≤n≤106,−100≤v[i]≤100) ( 3 ≤ n ≤ 10 6 , − 100 ≤ v [ i ] ≤ 100 )
Output
如果存在合法解则输出两个点的编号,否则输出 −1 − 1
Sample Input
6
2 4
0 5
4 2
2 1
1 1
4 2
Sample Output
1 4
Solution
如果总点权和不能被 3 3 整除则显然无解,否则树形统计以每个点为根的子树的点权和,只要某棵子树的点权和为总点权和的三分之一则记录下来,且从整棵树中划掉这棵子树,最后记录的点数只要有两个则有解,否则无解
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 1111111
int n,sum,v[maxn],dp[maxn],fa[maxn];
vector<int>g[maxn],ans;
void dfs(int u)
{
dp[u]=v[u];
for(int i=0;i<g[u].size();i++)
{
int v=g[u][i];
dfs(v);
if(dp[v]==sum)ans.push_back(v);
else dp[u]+=dp[v];
}
}
int main()
{
while(~scanf("%d",&n))
{
ans.clear();
for(int i=1;i<=n;i++)g[i].clear();
int root;
sum=0;
for(int i=1;i<=n;i++)
{
scanf("%d%d",&fa[i],&v[i]);
if(fa[i]==0)root=i;
sum+=v[i];
g[fa[i]].push_back(i);
}
if(sum%3!=0)
{
printf("-1\n");
continue;
}
sum/=3;
dfs(root);
if(ans.size()<2)printf("-1\n");
else printf("%d %d\n",ans[0],ans[1]);
}
return 0;
}