首页 > 代码库 > CodeForces 767C Garland
CodeForces 767C Garland
树形$dp$。
先看权值之和是否为$3$的倍数,如果不是则一定无解。
如果是$3$的倍数,可以分两次去切。每次一个节点,要求这个节点不是根,并且的子树权值和为$sum/3$,又要是深度最深的。
找不到两个依然是无解,否则就有解。
#pragma comment(linker, "/STACK:1024000000,1024000000")#include<cstdio>#include<cstring>#include<cmath>#include<algorithm>#include<vector>#include<map>#include<set>#include<queue>#include<stack>#include<ctime>#include<iostream>using namespace std;typedef long long LL;const double pi=acos(-1.0);void File(){ freopen("D:\\in.txt","r",stdin); freopen("D:\\out.txt","w",stdout);}template <class T>inline void read(T &x){ char c = getchar(); x = 0; while(!isdigit(c)) c = getchar(); while(isdigit(c)) { x = x * 10 + c - ‘0‘; c = getchar(); }}int n,root,c[1000010],p[1000010],sum,ans1,ans2;vector<int >v[1000010];int flag1,flag2;void dfs1(int x){ c[x]=p[x]; for(int i=0;i<v[x].size();i++) { dfs1(v[x][i]); if(flag1) return ; c[x]+=c[v[x][i]]; } if(x!=root&&c[x]==sum/3) { flag1=1; ans1=x; }}void dfs2(int x){ c[x]=p[x]; for(int i=0;i<v[x].size();i++) { if(v[x][i]==ans1) continue; dfs2(v[x][i]); if(flag2) return ; c[x]+=c[v[x][i]]; } if(x!=root&&c[x]==sum/3) { flag2=1; ans2=x; }}int main(){ scanf("%d",&n); for(int i=1;i<=n;i++) { int fa; scanf("%d%d",&fa,&p[i]); sum=sum+p[i]; if(fa==0) root=i; else v[fa].push_back(i); } if(sum%3!=0) { printf("-1\n"); return 0; } dfs1(root); if(flag1==0) { printf("-1\n"); return 0; } memset(c,0,sizeof c); dfs2(root); if(flag2==0) { printf("-1\n"); return 0; } else { printf("%d %d\n",ans1,ans2); } return 0;}
CodeForces 767C Garland
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。