题目链接:http://codeforces.com/contest/767/problem/C
问能否将一棵带点权的书分成点权$3$块,求任意方案。
其实考虑一棵以$x$为根的子树权值为${frac{1}{3}sum val[i]}$之后它就一定要作为单独的一块了,那么DFS一遍即可。
1 #include<iostream> 2 #include<cstdio> 3 #include<algorithm> 4 #include<vector> 5 #include<cstdlib> 6 #include<cmath> 7 #include<cstring> 8 using namespace std; 9 #define maxn 1000010 10 #define llg long long 11 #define yyj(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout); 12 llg n,m,ans[maxn],tot,val[maxn],size[maxn],sum,dad[maxn],root; 13 vector<llg>a[maxn]; 14 15 void dfs(llg x,llg fa) 16 { 17 if (tot==2) return ; 18 llg w=a[x].size(),v; 19 size[x]=val[x]; 20 for (llg i=0;i<w;i++) 21 { 22 if (tot==2) return ; 23 v=a[x][i]; 24 if (v==fa) continue; 25 if (tot==2) return ; 26 dfs(v,x); 27 if (tot==2) return ; 28 size[x]+=size[v]; 29 } 30 if (size[x]==sum) {ans[++tot]=x; size[x]=0;} 31 if (tot==2) return ; 32 } 33 34 void init() 35 { 36 cin>>n; 37 for (llg i=1;i<=n;i++) 38 { 39 scanf("%I64d%I64d",&dad[i],&val[i]); 40 if (dad[i]==0) root=i; 41 a[dad[i]].push_back(i); 42 sum+=val[i]; 43 } 44 } 45 46 int main() 47 { 48 // yyj("C"); 49 init(); 50 if (sum%3!=0) {cout<<-1; return 0;} 51 sum/=3; 52 dfs(root,0); 53 sort(ans+1,ans+tot+1); 54 for (llg i=1;i<=n;i++) 55 if (ans[i]==root) 56 { 57 cout<<-1; 58 return 0; 59 } 60 if (tot>=2) {for (llg i=1;i<=tot;i++) cout<<ans[i]<<" "; }else cout<<-1; 61 return 0; 62 }