题目链接:点击查看
题目大意:给出一棵有根树,每个节点都有一个权值,代表的是在其子树中有多少个节点的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;
}