2021牛客多校第十场补题 F-Train Wreck

题目链接

题目大意

模拟一个栈 告诉你火车进出的顺序 然后给出你每一辆火车的颜色 你需要保证 每次进栈时的栈是不同的 如果可以 输出颜色的进出顺序 如果不可以输出no

题目思路

先将所有出现过的颜色按照数量进行排序 存入优先队列中

将给出的括号顺序当作一棵树 一个括号里有多少个括号代表这个括号有多少个子节点

举例子

()(())()

相当于一棵这样的树

 (()())（）

 相当于一棵这样的树

对于每个节点的子节点 他们必须是不同的

如果相同 那么就会出现相同的序列 不符合题意

建完树之后 从根节点开始dfs

对于这个节点的所有子节点 进行判断 如果小于当前的 优先队列大小代表着 不能让所有子节点保持不同

就return flag=0

否则 根据当前节点所有的子节点数量k  对优先队列的前k个排头减1后再插入优先队列

注意不能立即插入优先队列 要将k个全部提出之后再进行插入操作 防止重复使用

 

当时做这道题的时候时间有点紧三点半才来实验室 非常可惜思路都想对了但是在建树的地方卡住

当时只考虑了 “（ ”的情况 其实对于“ ）”也需要回溯到上一级进行搜索 而不是简单的直接return

#include<bits/stdc++.h>
using namespace std;
const int maxn=3e6+10;
int n;
string s;
int a[maxn];
int ns[maxn];
struct node{
	int ty,num;
	bool operator <(const node &x)const 
	{
		return  x.num>num;
	}
}col[maxn];
priority_queue<node >q; 
vector<int>vec[maxn];
int tot=0;
int now=0; 
int f[maxn];
void dfs(int u,int fa)//建树 
{
	if(s[now]=='(')
	{
		vec[u].push_back(++tot);
		now++;
		f[tot]=u;
		dfs(tot,u);//要是（就查找子节点 
	}
	else if(s[now]==')')
	{
		now++;
		dfs(fa,f[fa]);//要是）就返回父节点继续查找 
	}
	return ;
}
int flag=1;
void dfs2(int now)
{
	if(flag==0)return ;
	if(vec[now].size()>q.size())
	{
		flag=0;
		return ;
	}
	if(flag==0)return ;
	queue<node >qq;//先用一个队列存下来需要--的元素 
	for(int i=0;i<vec[now].size();i++)
	{
		
		node t=q.top();
		q.pop();
		t.num--;
		ns[vec[now][i]]=t.ty;
		if(t.num==0)continue;
		qq.push(t);
	}
	while(!qq.empty())//最后插入这些元素
	{
		q.push(qq.front());
		qq.pop();
	}
	for(int i=0;i<vec[now].size();i++)
	{
		dfs2(vec[now][i]);//逐个子节点搜 
	}
	if(flag==0)return ;
}
int main()
{
	cin>>n;
	cin>>s;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
		col[a[i]].num++;
		col[a[i]].ty=a[i];
	}

	for(int i=1;i<=n;i++)
	{
		if(col[i].num!=0)q.push(node{i,col[i].num});
	}
	f[0]=-1;
	dfs(0,-1); 
	dfs2(0);
	if(flag)
	{
		cout<<"YES"<<endl;
		for(int i=1;i<=n;i++)cout<<ns[i]<<" ";
	}
	else cout<<"NO"<<endl;
	return 0;
}
/*
8
(()())((())()())
1 1 2 2 3 3 4 1
*/