HDU 5818 多校第七场 1010 Joint Stacks （线段树）

传送门：http://acm.hdu.edu.cn/showproblem.php?pid=5818
题解说这题有O(n)的做法，并没有想到，只想到线段树的。
直接把两个栈同时在一颗线段树里操作，标记一下每个区间A有几个，B有几个，然后merge的时候区间更新一下，相互翻转一下个数。每次查询的时候，实际上就是查询最右边的A或者最右边的B，pop掉的空位就留着不要了。
复杂度O(nlogn)，并不是很好。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <cstdlib>
#include <cctype>
#include <string>
#include <iostream>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <ctime>
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
#define pb push_back
#define mp make_pair
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define calm (l+r)>>1
const int INF = 2139062143;

const int maxn=1e5+10;
int n,Case;
int num[maxn];
struct Seg{
    int A[maxn<<2],B[maxn<<2],tag[maxn<<2];
    inline void pushup(int rt){
        A[rt]=A[rt<<1]+A[rt<<1|1];
        B[rt]=B[rt<<1]+B[rt<<1|1];
    }
    inline void pushdown(int rt){
        if(tag[rt]==1){
            A[rt<<1]+=B[rt<<1];
            B[rt<<1]=0;
            A[rt<<1|1]+=B[rt<<1|1];
            B[rt<<1|1]=0;
            tag[rt<<1]=tag[rt<<1|1]=tag[rt];
            tag[rt]=0;
        }
        else if(tag[rt]==2){
            B[rt<<1]+=A[rt<<1];
            A[rt<<1]=0;
            B[rt<<1|1]+=A[rt<<1|1];
            A[rt<<1|1]=0;
            tag[rt<<1]=tag[rt<<1|1]=tag[rt];
            tag[rt]=0;
        }
    }
    void build(int l,int r,int rt){
        A[rt]=B[rt]=tag[rt]=0;
        if(l==r)return;
        int m=calm;
        build(lson);
        build(rson);
    }
    void insert(int l,int r,int rt,int c,int x,int pos){
        if(l==r){
            if(c==0)A[rt]++;
            else B[rt]++;
            num[l]=x;
            //printf("insert %d into %d\n",x,l);
            return;
        }
        pushdown(rt);
        int m=calm;
        if(pos<=m){
            insert(lson,c,x,pos);
        }
        else{
            insert(rson,c,x,pos);
        }
        pushup(rt);
    }
    int query(int l,int r,int rt,int c){
        if(l==r){
            if(c==0)A[rt]--;
            else B[rt]--;
            int t=num[l];
            num[l]=0;
            return t;
        }
        pushdown(rt);
        int m=calm,ans=0;
        if(c==0){
            if(A[rt<<1|1])ans=query(rson,c);
            else ans=query(lson,c);
        }
        else{
            if(B[rt<<1|1])ans=query(rson,c);
            else ans=query(lson,c);
        }
        pushup(rt);
        return ans;
    }
    void merge(int c){
        tag[1]=c;
        if(c==1){
            A[1]+=B[1];B[1]=0;
        }
        else{
            B[1]+=A[1];A[1]=0;
        }
        //printf("A=%d B=%d,tag[1]=%d\n",A[1],B[1],tag[1]);
    }
}tree;
int main(){
    //freopen("D://input.txt","r",stdin);
    while(scanf("%d",&n)!=EOF&&n){
        tree.build(1,n,1);
        int now=0;//将要插入的位置
        printf("Case #%d:\n",++Case);
        for(int i=1;i<=n;i++){
            char s[10];scanf("%s",s);
            if(s[0]=='p'){
                char c[3];scanf("%s",c);
                if(s[1]=='u'){
                    int x;scanf("%d",&x);
                    if(c[0]=='A'){
                        tree.insert(1,n,1,0,x,++now);
                    }
                    else tree.insert(1,n,1,1,x,++now);
                }
                else{
                    if(c[0]=='A')printf("%d\n",tree.query(1,n,1,0));
                    else printf("%d\n",tree.query(1,n,1,1));
                }
            }
            else if(s[0]=='m'){
                char c[3];
                scanf("%s",c);scanf("%s",c);
                if(c[0]=='A'){//B A
                    tree.merge(2);
                }
                else{//A B
                    tree.merge(1);
                }
            }
        }
    }
    return 0;
}