codeforces 767C - Garland

题目链接：http://codeforces.com/contest/767/problem/C

---

问能否将一棵带点权的书分成点权$3$块，求任意方案。

其实考虑一棵以$x$为根的子树权值为${\frac{1}{3}\sum val[i]}$之后它就一定要作为单独的一块了，那么DFS一遍即可。

---

 #include<iostream>
 #include<cstdio>
 #include<algorithm>
 #include<vector>
 #include<cstdlib>
 #include<cmath>
 #include<cstring>
 using namespace std;
 #define maxn 1000010
 #define llg long long
 #define yyj(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout);
 llg n,m,ans[maxn],tot,val[maxn],size[maxn],sum,dad[maxn],root;
 vector<llg>a[maxn];

 void dfs(llg x,llg fa)
 {
     if (tot==) return ;
     llg w=a[x].size(),v;
     size[x]=val[x];
     for (llg i=;i<w;i++)
     {
         if (tot==) return ;
         v=a[x][i];
         if (v==fa) continue;
         if (tot==) return ;
         dfs(v,x);
         if (tot==) return ;
         size[x]+=size[v];
     }
     if (size[x]==sum) {ans[++tot]=x; size[x]=;}
     if (tot==) return ;
 }

 void init()
 {
     cin>>n;
     for (llg i=;i<=n;i++)
     {
         scanf("%I64d%I64d",&dad[i],&val[i]);
         if (dad[i]==) root=i;
         a[dad[i]].push_back(i);
         sum+=val[i];
     }
 }

 int main()
 {
 //    yyj("C");
     init();
     if (sum%!=) {cout<<-; return ;}
     sum/=;
     dfs(root,);
     sort(ans+,ans+tot+);
     for (llg i=;i<=n;i++)
         if (ans[i]==root)
         {
             cout<<-;
             return ;
         }
     if (tot>=) {for (llg i=;i<=tot;i++) cout<<ans[i]<<" "; }else cout<<-;
     return ;
 }