用两个栈模拟,并保存每个点的时间戳。每次合并的时候记录合并时的时间戳mcnt和此时的topa和topb记做ta、tb。
每次pop的时候,如果栈的top的时间戳大于mcnt,则普通地pop,否则就在两个栈ta和tb下面找时间戳最大且还没pop掉的。然后用bj[时间戳]来标记已经pop了。
#include <cstdio> #include <cstring> #define N 100005 using namespace std; struct node{ int id,v; }a[N],b[N]; int bj[N],n,topa,topb,cnt,ele,ta,tb,mcnt,cas; char op[50],st; void popc(){ while(bj[a[ta].id])ta--; while(bj[b[tb].id])tb--; if(a[ta].id>b[tb].id){ printf("%d\n",a[ta].v); bj[a[ta].id]=1; }else { printf("%d\n",b[tb].v); bj[b[tb].id]=1; } } int main() { while(~scanf("%d",&n)&&n){ printf("Case #%d:\n",++cas); topa=topb=cnt=mcnt=0; memset(bj,0,sizeof bj); for(int i=1;i<=n;i++){ scanf("%s %c",op,&st); if(op[1]==‘u‘){ scanf("%d",&ele); if(st==‘A‘) a[++topa]=(node){++cnt,ele}; else b[++topb]=(node){++cnt,ele}; }else if(op[1]==‘o‘){ if(st==‘A‘){ if(a[topa].id>mcnt){ printf("%d\n",a[topa].v); bj[a[topa].id]=1; topa--; } else popc(); }else{ if(b[topb].id>mcnt){ printf("%d\n",b[topb].v); bj[b[topb].id]=1; topb--; } else popc(); } }else if(op[1]==‘e‘){ scanf(" %c",&st); mcnt=cnt; ta=topa; tb=topb; } } } return 0; }
wa了好几发,结果原因是 “if(a[topa].id>mcnt)”写的是>=,为什么等于不可以呢,因为popc函数里没有top--,这样下一次top的值可能已经pop掉了。那如果在popc里面top--呢,也不行,因为pop栈a的ta时,可能在pop栈b,这样to pa--的话就错了。这题用优先队列也很方便。
时间: 2024-12-15 17:15:59