CodeForces - 1287D Numbers on Tree(dfs+stl)

题目链接：点击查看

题目大意：给出一棵有根树，每个节点都有一个权值，代表的是在其子树中有多少个节点的val比他小，现在要求根据每个点的权值构造出1~n的val数列

题目分析：因为数据量不大，n只有2e3，所以直接dfs乱搞就行了

这里学了一波在递归里用exit(0)直接退出，用起来蛮不错的

代码：

#include<iostream>
#include<cstdlib>
#include<string>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<climits>
#include<cmath>
#include<cctype>
#include<stack>
#include<queue>
#include<list>
#include<vector>
#include<set>
#include<map>
#include<sstream>
#include<unordered_map>
using namespace std;
    
typedef long long LL;
    
const int inf=0x3f3f3f3f;

const int N=2e3+100;

vector<int>node[N];

int c[N],ans[N];

vector<int>dfs(int u)
{
	vector<int>rank;
	for(auto v:node[u])
	{
		vector<int>temp=dfs(v);
		rank.insert(rank.end(),temp.begin(),temp.end());
	}
	if(c[u]>rank.size())
	{
		puts("NO");
		exit(0);
	}
	rank.insert(rank.begin()+c[u],u);
	return rank;
}
 
int main()
{
//  freopen("input.txt","r",stdin);
//  ios::sync_with_stdio(false);
	int n,root;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		int fa;
		scanf("%d%d",&fa,c+i);
		if(fa==0)
			root=i;
		else
			node[fa].push_back(i);
	}
	vector<int>rank=dfs(root);
	for(int i=0;i<rank.size();i++)
		ans[rank[i]]=i+1;
	puts("YES");
	for(int i=1;i<=n;i++)
		printf("%d ",ans[i]);


    
    
    
    
 
        
        
        
         
        
    return 0;
}