F-Train Wreck_2021牛客暑期多校训练营10

F-Train Wreck

题目大意

链接：https://ac.nowcoder.com/acm/contest/11261/F
来源：牛客网
有n辆涂着颜色的火车,现在开进一条一边封住的轨道,每次将一辆火车推入或将顶部的火车推出,就像一个栈.在一辆火车被推入后,所有火车颜色构成的序列不能和之前出现过的序列一样.就如下图所示.

先输入n
第二行输入一个长度为$2\cdot n$的字符串.’('代表入栈, ')'代表出栈.
第三行输入n个整数,第i个代表了第i辆火车的颜色(颜色会重复).

思路

既然要模拟栈,那么就可以根据出入栈的顺序构建一颗树来模拟.
将每个代表在栈为空时压入栈的元素连向一个虚拟节点0.
然后为每个节点涂上颜色.由于不能有重复序列,一个节点的子节点不能有两个点颜色相同.
再用优先队列来进行维护,pair<int,int>的第一个数代表了该颜色还剩余几辆火车,第二个数代表火车的颜色.用贪心的思想,每次将最多颜色给找到的节点涂上,具体见代码.

代码实现

#include<iostream>
#include<cstdio>
#include<vector>
#include<queue>
#include<utility>
using namespace std;
priority_queue<pair<int,int> > h;
vector<int> v[1000010];
int n,num[1000010],ans[1000010];
char ch[2000010];
void check(vector<int> a){
	vector<pair<int,int> > tmp;
	for(int i=0;i<a.size();i++){
		if(h.empty()){
			printf("NO\n");
			exit(0);
		}
		pair<int,int> pr=h.top();
		h.pop();
		ans[a[i]]=pr.second;
		tmp.push_back({pr.first-1,pr.second});
	}
	for(int i=0;i<tmp.size();i++){
		pair<int,int> a=tmp[i];
		h.push(a);
	}
}
int main(){
	scanf("%d",&n);
	scanf(" %s",ch+1);
	for(int i=1;i<=n;i++){
		int x;
		scanf("%d",&x);
		num[x]++;
	}
	for(int i=1;i<=n;i++){
		if(num[i]){
			h.push({num[i],i});
			num[i]=0;
		}
	}
	int tot=0,top=0;
	for(int i=1;i<=2*n;i++){
		if(ch[i]=='('){
			v[top].push_back(++tot);
			top++;
			v[top].clear();
		}
		else{
			check(v[top--]);
		}
	}
	check(v[top]);
	printf("YES\n");
	for(int i=1;i<=n;i++)
		printf("%d ",ans[i]);
}