F-Train Wreck

题目大意

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

先输入n
第二行输入一个长度为2⋅n2\cdot n2⋅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]);
}

F-Train Wreck_2021牛客暑期多校训练营10相关推荐

“蔚来杯“2022牛客暑期多校训练营10，签到题HFIE
题号标题已通过代码通过率团队的状态 A Everlasting Transeunt 点击查看 6/42 B Fall Guys-Perfect Match 点击查看 6/115 C Magic ...
2021牛客暑期多校训练营10，签到题FH
题号标题已通过代码通过率团队的状态 A Browser Games 点击查看 93/1498 未通过 B Child's play 点击查看 0/8 未通过 C Dance Party 点击查 ...
“蔚来杯“2022牛客暑期多校训练营10补题
H Wheel of Fortune 题意有两个人在打炉石,一方转动了尤格萨隆的命运之轮触发了炎爆选项.双方英雄的血量分别为A和B,双方场面的血量分别为ai和bi,问A获胜的概率. 思路将代码分 ...
2021牛客暑期多校训练营
2021牛客暑期多校训练营1 A- Alice and Bob /* 博弈问题两堆石头,每人每次从其中一堆拿k(k>0)个, 同时从另一堆拿s*k个(s>=0),不能进行操作的即失败 * ...
2019牛客暑期多校训练营（第五场）C generator 2 (BSGS)
2019牛客暑期多校训练营(第五场)C generator 2 思路 x0=x0x_0 = x_0x0=x0 x1=a∗x0∗bx_1 = a * x_0 * bx1=a∗x0∗b x2=a∗ ...
2021牛客暑期多校训练营9
2021牛客暑期多校训练营9 题号题目知识点 A A Math Challenge B Best Subgraph C Cells D Divide-and-conquer on Tree E E ...
2021牛客暑期多校训练营5
2021牛客暑期多校训练营5 题号题目知识点 A Away from College B Boxes 概率 C Cheating and Stealing D Double Strings 线性d ...
2021牛客暑期多校训练营4
2021牛客暑期多校训练营4 题号题目知识点 A Course B Sample Game C LCS D Rebuild Tree E Tree Xor 思维+线段树 F Just a joke ...
2021牛客暑期多校训练营3
2021牛客暑期多校训练营3 题号题目知识点 A Guess and lies B Black and white C Minimum grid 二分图匹配 D Count E Math 数论+打 ...
2021牛客暑期多校训练营2
2021牛客暑期多校训练营2 题号题目知识点 A Arithmetic Progression B Cannon C Draw Grids D Er Ba Game E Gas Station F ...

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

F-Train Wreck

题目大意

思路

代码实现

F-Train Wreck_2021牛客暑期多校训练营10相关推荐

最新文章

热门文章