[codeforces 1287D] Numbers on Tree 子树节点数

[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;
}