原题链接
n(<=1e5)个物品,每个物品具有各自的颜色(可能相同),给出一串表示出入栈顺序的字符串,求出入栈顺序对应物品是什么,要求每次入栈形成栈中的物品颜色排列在之前从未出现过。
分析:
可以将问题转化成建立一棵树,使每个节点的下一层的所有子节点的颜色都不相同。入栈即是向下延伸一个节点,出栈则是回到上一个节点。
根据贪心,每次插入节点,应该优先插入当前颜色出现次数最多的、在其他子节点未出现过的颜色,故用堆维护颜色出现次数,然后不断更新即可。
char str[maxv];
int cnt[maxv],ans[maxv];
priority_queue<pair<int,int> > q;
vector<pair<int,int> > temp;
vector<int> g[maxv],all;
int main(){
int n,k,ind=0;
scanf("%d%s",&n,str+1);
for(int i=1;i<=n;i++) scanf("%d",&k),cnt[k]++;
for(int i=1;i<=n;i++) if(cnt[i]) q.push({cnt[i],i});
all.push_back(0);
for(int i=1;i<=n*2;i++)
if(str[i]=='(') g[all.back()].push_back(++ind),all.push_back(ind);
else all.pop_back();
for(int i=0;i<=n;i++){
if(g[i].size()>q.size()){printf("NO");return 0;}
temp.clear();
for(int j=0,y;j<g[i].size();j++){
temp.push_back(q.top()),ans[g[i][j]]=q.top().second;
q.pop();
}
for(int j=0,y;j<temp.size();j++)
if(temp[j].first>=2) q.push({temp[j].first-1,temp[j].second});
}
printf("YES\n");
for(int i=1;i<=n;i++) printf("%d ",ans[i]);
return 0;
}