2021牛客多校#10 F-Train Wreck

原题链接

			https://ac.nowcoder.com/acm/contest/11261/F

题目大意

有一个旧火车站，其只有一个死胡同轨道，可以视为一个栈。有$n(1\le n\le 10^6)$辆彩色火车进站，第$i$辆火车的颜色是$k_i(1\le k_i\le n)$，进出栈顺序以定，现在你需要为每次入栈操作分配一辆火车，是的入栈时栈中其他火车颜色序列唯一。

题解

我们先模拟一下这种火车入栈的操作，不难发现其操作可以以树的方式出现，如下图：

然后我们可以发现，当一个序列唯一时，以某节点为根节点的子节点中任意两点颜色不同（即剩余颜色数量不小于子节点数量）此时有解。

所以我们只需要通过优先队列维护不同颜色节点数从大到小（因为颜色数越多，方案越不容易重复）并给所有节点染色即可。

若给子节点染色时，颜色数小于子节点数，无解。

参考代码

#include<bits/stdc++.h>
#define pb push_back
#define pq priority_queue
using namespace std;
const int N=2e6+5;
int n;
int tot,top,ans[N],c[N];
pq<pair<int,int> >q;
vector<int>v[N];
void w(vector<int>a)
{
	vector<pair<int,int> >tmp;
	for(int i=0;i<a.size();i++)
	{
		if(q.empty())
		{
			puts("NO");
			exit(0);
		}
		int fi=q.top().first,se=q.top().second;
		q.pop();
		ans[a[i]]=se;
		tmp.pb({fi-1,se});//每次染色后将这种颜色的数量-1
	}
	for(int i=0;i<tmp.size();i++)
	{
		int fi=tmp[i].first,se=tmp[i].second;
		if(fi>0)q.push({fi,se});//若一种颜色已经用完，不存入队列
	}
}
int main()
{
	scanf("%d",&n);
	string st;
	cin>>st;
	for(int i=1;i<=n;i++){int x;scanf("%d",&x);c[x]++;}
	for(int i=1;i<=n;i++)
		if(c[i]){q.push({c[i],i});c[i]=0;}//初始化优先队列
	tot=0;top=0;
	for(int i=0;i<=2*n-1;i++)
	{
		if(st[i]=='(')
		{
			v[top].pb(++tot);//存入需要染色的节点
			top++;
			v[top].clear();
		}
		else
			w(v[top--]);
	}
	w(v[top]);
	puts("YES");
	for(int i=1;i<=n;i++)
		printf("%d ",ans[i]);
}