Numbers on Tree(DFS)

最新推荐文章于 2024-08-06 17:42:20 发布

WYW___

最新推荐文章于 2024-08-06 17:42:20 发布

阅读量205

点赞数

文章标签：深度优先算法图论

本文链接：https://blog.csdn.net/m0_64158084/article/details/127316839

版权

Problem - 1287D - Codeforces

题意:给你每个节点的父节点和他子树中有多少比他小的点,让你给这个数赋权满足上述条件

原文链接：https://blog.csdn.net/jziwjxjd/article/details/106177186

首先,构造千万不要想复杂了,尽量往简单的想

我们构造的 a [ i ] 应该互不相同最好 , 因为相同不好选择

比其中一个大就比很多个大 , 不好确定排名

既然如此 , 我就规定最终 a [ i ] 取 [ 1 , n ]

这样一来根就被唯一确定是 , a [ r o o t ] = c [ r o o t ] + 1

然后 d f s 遍历下去 , 每次找到剩余没有用过的数字中第 c [ i ] + 1 大的数

如果某个顶点的所有子节点数加起来都小于 c [ i ] , 无解

某次为顶点找第 c [ i ] + 1 大的数没找到 , 无解

#include<iostream>
#include<vector>
using namespace std;
int n;
int tem[2005];
int c[2005];
vector<int> a[2005];
int f = 0;
int Size[2005];
int ans[2005];
int find(int x)
{
	int t = 0;
	for(int i = 1; i <= n ; i++)
	{
		if(tem[i] == 0)
		continue;
		t ++;
		if(t == x)
		{
			tem[i] = 0;
			return i;
		}
	}
	return -1;
}
void dfs(int x)
{
	Size[x] = 1;
	ans[x] = find(c[x] + 1);
	if(ans[x] == -1)
    f = -1;
	for(auto v:a[x])
	{
		dfs(v);
		Size[x] += Size[v];
	}
	if(Size[x] - 1 < c[x])
	{
		 f = -1;
	}
}
int main()
{
	scanf("%d",&n);
	for(int i = 1; i <= n; i++)
	tem[i] = 1;
	int root = 0;
	for(int i = 1; i <= n; i++)
	{
		int l;
		scanf("%d%d",&l,&c[i]);
		a[l].push_back(i);
		if(l == 0)
		{
			root = i;
		}
	}
	dfs(root);
	if(f == -1)
	printf("NO");
	else
	{
		printf("YES\n");
		for(int i = 1; i <= n; i++)
		printf("%d ",ans[i]);
	}
}