題目鏈接：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 }

?

轉載于:https://www.cnblogs.com/Dragon-Light/p/6414187.html

與50位技術專家面對面20年技術見證，附贈技術全景圖

總結

以上是生活随笔為你收集整理的codeforces 767C - Garland的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。