迫真小游戏
题目链接:https://cometoj.com/contest/46/problem/C?problem_id=2028
题目描述
H君喜欢在阳台晒太阳,闲暇之余他会玩一些塔防小游戏。
H君玩的小游戏可以抽象成一棵 nn 个节点的有根树,树以 11 为根,每个点的深度定义为其到根的简单路径上的点数(根的深度为 11)。
H君有 nn 个干员,H君会按照某种顺序把她们部署到树的每一个节点上,使得每个节点上恰好有一个干员。由于游戏的机制,他们对每个节点 ii 都给出了个限制参数 a_iai ,要求H君在第 ii 个节点部署干员之前,所有深度 > a_i>ai 的节点上不能有干员。同时游戏为了让玩家过关,保证了 a_iai 大于等于点 ii 的深度。
H君将每一次部署干员的节点按顺序写在纸上,形成了一个 1 \dots n1…n 的排列,H君为了获得更多的奖励,想要最小化这个排列的字典序。
我们认为排列 c_1,c_2..c_nc1,c2..cn 的字典序比排列 d_1,d_2..d_nd1,d2..dn 的字典序小,当且仅当 c, dc,d 不完全相同且存在一个下标 ii,满足 c_i < d_ici<di 且对于所有 1 \le j < i1≤j<i 的 jj 都有 c_j = d_jcj=dj 。
输入描述
第一行一个数 nn 。
接下来 n - 1n−1 行,每行两个数 x, yx,y 表示树上的一条边 。
最后一行 nn 个数,表示 a_iai 。
数据范围:
1\le n \le 5 \times 10^5, 1 \le a_i \le n1≤n≤5×105,1≤ai≤n。
输出描述
第一行 nn 个数,表示字典序最小的排列。
样例输入 1
5 1 5 5 3 1 4 4 2 1 3 3 3 2
样例输出 1
1 4 5 2 3
样例输入 2
10 1 7 7 8 7 2 8 9 7 6 2 4 9 5 8 10 6 3 5 3 4 4 5 3 5 3 5 4
题目是中文题,题意就不描述了
思路:先dfs一遍树把所有节点的高度给记录下来,然后再根据题意维护一个最小堆即可,具体看代码。
代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#define LL long long
using namespace std;
const int maxn=5e5+100;
vector<int>mp[maxn];
int d[maxn];
int a[maxn];
vector<int>de[maxn];
int Maxlen=-1;
void dfs(int u,int fa,int len)
{
de[len].push_back(u);
Maxlen=max(len,Maxlen);
for(int i=0;i<mp[u].size();i++)
{
int v=mp[u][i];
if(v==fa)
{
continue;
}
dfs(v,u,len+1);
}
return ;
}
int main()
{
int n;
scanf("%d",&n);
for(int i=1;i<n;i++)
{
int u,v;
scanf("%d%d",&u,&v);
mp[u].push_back(v);
mp[v].push_back(u);
}
for(int i=1;i<=n;i++)
{
int gg;
scanf("%d",&gg);
d[gg+1]++;
a[i]=gg+1;
}
dfs(1,0,1);
priority_queue<int,vector<int>,greater<int> >Q;
Q.push(1);
int tt=2;
int p=1;
vector<int>ans;
while(!Q.empty())
{
p=Q.top();
Q.pop();
d[a[p]]--;
ans.push_back(p);
while(tt<=Maxlen&&d[tt]==0)
{
for(int i=0;i<de[tt].size();i++)
{
p=de[tt][i];
Q.push(p);
}
tt++;
}
}
int si=ans.size();
for(int i=0;i<si;i++)
{
if(i!=si-1)
{
printf("%d ",ans[i]);
}
else
{
printf("%d\n",ans[i]);
}
}
return 0;
}