[codeforces 1287D] Numbers on Tree 子树节点数
总目录详见https://blog.csdn.net/mrcrack/article/details/103564004
在线测评地址https://codeforces.com/contest/1287/problem/D
Problem | Lang | Verdict | Time | Memory |
---|---|---|---|---|
D - Numbers on Tree | GNU C++11 | Accepted | 31 ms | 0 KB |
思路同https://blog.csdn.net/weixin_43826249/article/details/103855447
首先证明:n个节点一定可以用1-n的数字去放。
因为1-n的数字都是不同的,那么不会有相对大小的问题,不会有节点之间相互影响而导致答案错误。放相同值的节点也就是少一个差值,我们完全可以用放的方式去避免,比如小的放在大的上面。
其次证明:如果节点的C值小于节点的size,那么一定存在这种方案。
我们从顶点开始放,因为我们放1到n的数字,那么有多少个数字没放是一定的,所以我们相当于是从没放的数字当中找第C+1大的数字放上去,然后标记放的数字,向下递归。一定满足。
#include <stdio.h>
#define N 2010
int p,c[N],size[N],n,head[N],tot,num[N],vis[N];
struct node{
int to,next;
}e[N];
void add_edge(int u,int v){
tot++,e[tot].to=v,e[tot].next=head[u],head[u]=tot;
}
void dfs_size(int x){
int v,b;
size[x]=1;
for(b=head[x];b;b=e[b].next){
v=e[b].to;
dfs_size(v);
size[x]+=size[v];
}
}
void dfs(int x){
int i,cnt=0,b;
for(i=1;i<=n;i++){
if(!vis[i])cnt++;
if(c[x]+1==cnt){
num[x]=i,vis[i]=1;
break;
}
}
for(b=head[x];b;b=e[b].next)dfs(e[b].to);
}
int main(){
int i,rt;
scanf("%d",&n);
for(i=1;i<=n;i++){
scanf("%d%d",&p,&c[i]);
if(!p)rt=i;
else add_edge(p,i);
}
dfs_size(rt);
for(i=1;i<=n;i++)
if(c[i]+1>size[i]){
printf("NO");
return 0;
}
dfs(rt);
printf("YES\n");
for(i=1;i<=n;i++)printf("%d ",num[i]);
return 0;
}