Codeforces Round #612 B. Numbers on Tree【思维】

题目链接

题意

给出一棵以 1 为根的树，每个结点有个两个权值 $p_i,a_i$ ，其中 $p_i$表示它的子树中 $a_i$ 比它小的结点数量，现在给出 $p_i$，求一种满足的$a_i$，不存在则输出NO

题解

不满足的情况为 $p_i$ 大于其子树结点数

对于满足的情况，一定存在一种 1~n 的排列符合，对于 $p_i$ ，$a_i$就是 1~n 里剩余的第 $p_i+1$ 个数，每次用完一个数就把它删了。

暴力找是 $n^2$ 的，我用了个平衡树降到 $nlogn$ 但好像时间差不多

#include<iostream>
#include<sstream>
#include<string>
#include<queue>
#include<map>
#include<unordered_map>
#include<set>
#include<vector>
#include<stack>
#include <utility>
#include<list>
#include<bitset>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<iomanip>
#include<time.h>
#include<random>
using namespace std;
#include<ext/pb_ds/priority_queue.hpp>
#include<ext/pb_ds/tree_policy.hpp>
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/hash_policy.hpp>
using namespace __gnu_pbds;
#include<ext/rope>
using namespace __gnu_cxx;

#define int long long
#define PI acos(-1.0)
#define eps 1e-9
#define lowbit(a) ((a)&-(a))
#define mid ((l+r)>>1)
#define mem(x,y) memset(x,y,sizeof x)

const int mod = 1e9+7;
int qpow(int a,int b){
	int ans=1;
	while(b){
		if(b&1)ans=(ans*a)%mod;
		a=(a*a)%mod;
		b>>=1;
	}
	return ans;
}
const int INF = 0x3f3f3f3f;
const int N = 1e6+10;
vector<int>g[N];
int rt,n,c[N],ans[N],sz[N],flag=1;
tree<int,null_type,less<int>,rb_tree_tag,tree_order_statistics_node_update>rb;
void dfs(int u){
	sz[u]=1;
	ans[u]=*rb.find_by_order(c[u]);//pbds yyds
	rb.erase(rb.find_by_order(c[u]));
	for(auto v:g[u])dfs(v),sz[u]+=sz[v];
	if(sz[u]-1<c[u])flag=0;
}
#define endl '\n'
signed main(){
	std::ios::sync_with_stdio(false);
	cin.tie(0),cout.tie(0);
	cin>>n;
	for(int i=1;i<=n;i++){
		int p; cin>>p>>c[i]; rb.insert(i);
		if(!p){rt=i; continue;}
		g[p].push_back(i); 
	}
	dfs(rt);
	if(!flag){cout<<"NO"<<endl; return 0;}
	cout<<"YES"<<endl;
	for(int i=1;i<=n;i++)cout<<ans[i]<<" ";
}