HDU 5818 2016多校賽第七場 數組模擬鏈表，來模擬棧

http://acm.hdu.edu.cn/showproblem.php?pid=5818

比較簡單巧妙的一個做法是引入一個新的棧C，每次合併的時候就把A和B合併到C上，然後把A和B都清空. push還是按正常做，pop注意當遇到要pop的棧爲空時，因爲題目保證不會對空棧進行pop操作，所以這時應直接改爲對C棧進行pop操作. 這樣做因爲保證每個元素最多隻在一次合併中被處理到，pop和push操作當然也是每個元素只做一次，所以總複雜度是O(N)的. 另一種做法是用鏈表來直接模擬，複雜度也是O(N)，但代碼量稍大一些.

將所有的未被刪點存入一個數組裏，用三個變量標記一下棧A，棧B，和公共棧，的棧頂。

#include<bits/stdc++.h>
using namespace std;
const int N=210000;
struct node{
    int pre;  ///左邊下標
    int next; ///右邊下標，即棧頂的第二個元素
    int val;
}p[N];
int num=0,n,head;
int top[2];
int a[2][N];
int toop;
int build(int pre,int next,int val){
    p[num].pre=pre;
    p[num].next=next;
    p[num].val=val;
    return num++;
}
void push(int ind,int d){
    int now = build(-1,-1,d);
    p[now].next=p[head].next;
    if(p[head].next!=-1)
        p[p[head].next].pre=now;
    p[now].pre=head;
    p[head].next=now;
    a[ind][++top[ind]]=now;
}
void pop(int ind){
    if(!top[ind]){
        printf("%d\n",p[toop].val);
        if(p[toop].next!=-1)
        p[p[toop].next].pre=p[toop].pre;
        p[p[toop].pre].next=p[toop].next;
        toop=p[toop].next;  ///棧頂的指針改變
    }
    else{
        int now=a[ind][top[ind]]; top[ind]--;
        printf("%d\n",p[now].val);
        if(p[now].next!=-1)
        p[p[now].next].pre=p[now].pre;
        p[p[now].pre].next=p[now].next;
    }
}

int main(){
    char s1[10],s[10],s2[19];
    int d;
    int cas=0;
    while(~scanf("%d",&n),n){
        top[0]=top[1]=0;
        printf("Case #%d:\n",++cas);
        num=0;
        head=build(-1,-1,-1);
        while(n--){
            scanf("%s",s);
            if(s[1]=='u'){
                scanf("%s%d",s1,&d);
                push(s1[0]-'A',d);
            }
            else if(s[1]=='o'){
                scanf("%s",s1);
                pop(s1[0]-'A');
            }
            else{
                scanf("%s%s",s1,s2);
                toop=p[head].next;  ///用來標記合併時下標，必須，防止以後可能會在另一個棧中存了數
                top[0]=top[1]=0;
            }
        }
    }
    return 0;
}