HDU 5818 2016多校赛第七场数组模拟链表，来模拟栈

最新推荐文章于 2019-08-09 10:00:57 发布

原创最新推荐文章于 2019-08-09 10:00:57 发布 · 409 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

栈和队列同时被 3 个专栏收录

21 篇文章

订阅专栏

链表

18 篇文章

订阅专栏

模拟

11 篇文章

订阅专栏

本文介绍了一种解决HDU 5818问题的有效方法，通过引入辅助栈C来合并栈A和B，并确保每个元素只被处理一次，从而实现O(N)的时间复杂度。此外还提供了一种使用链表模拟的方法。

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;
}