2021牛客多校10 F-Train Wreck（思维，贪心，堆）

赛场上有点思路，但是没码出来...

---

题目链接

题意

     有一个类似栈的车道，给定一个合法的括号序列，'('代表火车进站，')'代表出站，每个火车都有一种颜色，现在让选择合适的顺序，使得每次火车进站后，车道上的颜色序列各不相同。

思路

       观察一个样例()((())())(())，我们可以将每次火车进站后火车的数量记录一下，即(1)(1232)(12),可以发现数量为1的火车颜色不能相同，同样数量为2的火车颜色不能相同；但第二个括号中的2却可以与第三括号中的2颜色相同。

       这好像是树的遍历，我们抽象一个0结点，编号1的点为0的儿子，依次向下，发现只要对于每一个结点，其儿子的颜色各不相同即可，即判断当前不同颜色的数量是否大于等于结点儿子的数量，每个结点染何种颜色无所谓，贪心的每次取数量最多的颜色即可。

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

int tong[1000010];
priority_queue<pair<int,int> > q; //因为堆中要存每种颜色的数量及其编号，所以用了pair 
int ans[1000010],fa[1000010];
vector<int> vv[1000010];  //记录每个结点的儿子 
pair<int,int> p[1000010];  //用于染色时临时存储pair 

bool dfs(int now){
	int siz=vv[now].size();
	//将堆中的siz中颜色弹出，暂存到p中 
	for(int i=0;i<siz;i++){
		if(q.empty()) return false;
		p[i]=q.top(); q.pop();
	}
	//对每个儿子结点染色，颜色数量减一，操作之后若还有剩余放回堆中 
	for(int i=0;i<siz;i++){
		ans[vv[now][i]]=p[i].second;
		p[i].first--;
		if(p[i].first) q.push(p[i]);
	}
	//遍历儿子结点 
	for(int i=0;i<siz;i++) dfs(vv[now][i]);
	return true;
}

int main()
{
	ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
	int n,x;
	string s;
	cin>>n>>s;
	for(int i=1;i<=n;i++){
		cin>>x;
		tong[x]++;
	} 
	for(int i=1;i<=1000000;i++) if(tong[i]) q.push({tong[i],i});
	
	int now=0,cnt=0;
	for(int i=0;i<s.size();i++){
		//建树，结点的标号无所谓，从1开始存即可 
		if(s[i]=='('){
			vv[now].push_back(++cnt); fa[cnt]=now; now=cnt;
		}
		else now=fa[now]; 
	}
	
	bool f=dfs(0);
	
	if(f){
		cout<<"YES\n";
		for(int i=1;i<=n;i++) cout<<ans[i]<<" ";
	}
	else cout<<"NO\n";
	return 0;
}