BZOJ2141 排队

题目链接：http://www.lydsy.com/JudgeOnline/problem.php?id=2141

Description

排排坐，吃果果，生果甜嗦嗦，大家笑呵呵。你一个，我一个，大的分给你，小的留给我，吃完果果唱支歌，大家乐和和。红星幼儿园的小朋友们排起了长长地队伍，准备吃果果。不过因为小朋友们的身高有所区别，排成的队伍高低错乱，极不美观。设第i个小朋友的身高为hi，我们定义一个序列的杂乱程度为：满足ihj的(i,j)数量。幼儿园阿姨每次会选出两个小朋友，交换他们的位置，请你帮忙计算出每次交换后，序列的杂乱程度。为方便幼儿园阿姨统计，在未进行任何交换操作时，你也应该输出该序列的杂乱程度。

Input

第一行为一个正整数n，表示小朋友的数量；第二行包含n个由空格分隔的正整数h1,h2,…,hn，依次表示初始队列中小朋友的身高；第三行为一个正整数m，表示交换操作的次数；以下m行每行包含两个正整数ai和bi¬，表示交换位置ai与位置bi的小朋友。

Output

输出文件共m行，第i行一个正整数表示交换操作i结束后，序列的杂乱程度。

ZKW线段树套Treap，膜着CZL神犇的代码愣是找出了一堆bug

依然是十分二逼的写法（ZKW线段树连用指针都难）

注意query函数中的 if (l > r) return 0; 一句

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#define rep(i,l,r) for(int i=l; i<=r; i++)
#define clr(x,y) memset(x,y,sizeof(x))
#define travel(x) for(Edge *p=last[x]; p; p=p->pre)
using namespace std;
const int INF = 0x3f3f3f3f;
const int maxn = 20010;
inline int read(){
    int ans = 0, f = 1;
    char c = getchar();
    for(; !isdigit(c); c = getchar())
    if (c == '-') f = -1;
    for(; isdigit(c); c = getchar())
    ans = ans * 10 + c - '0';
    return ans * f;
}
int n,m,l,r,M,ans,h[maxn];
struct Node{
    Node *ch[2];
    int v,s,w,r;
    inline void maintain(){
        s = ch[0]->s + ch[1]->s + w;
    }
};
Node TREAP[525000],*pt = TREAP,*null = pt++;
inline Node *newnode(int x){
    pt->v = x; pt->s = pt->w = 1; pt->r = rand();
    pt->ch[0] = pt->ch[1] = null;
    return pt++;
}
inline void rotate(Node *&o,int d){
    Node *k = o->ch[d^1]; o->ch[d^1] = k->ch[d]; k->ch[d] = o;
    o->maintain(); k->maintain(); o = k;
}
struct Treap{
    Node *root;
    Treap(){
        root = null;
    }
    void ins(int x,Node *&p){
        if (p == null){
            p = newnode(x); return;
        }
        p->s++; if (x == p->v) p->w++;
        else{
            int d = x > p->v;
            ins(x,p->ch[d]);
            if (p->ch[d]->r < p->r) rotate(p,d^1);
        }
    }
    void del(int x,Node *&p){
        if (p == null) return;
        if (p->v == x){
            if (p->w > 1){
                p->s--; p->w--; return;
            }
            if (p->ch[0] != null && p->ch[1] != null){
                int d = p->ch[0]->r > p->ch[1]->r;
                rotate(p,d); del(x,p->ch[d]);
            }
            else p = p->ch[0] == null ? p->ch[1] : p->ch[0];
        }
        else{
            int d = x > p->v; p->s--;
            del(x,p->ch[d]);
        }
        if (p != null) p->maintain();
    }
    int get_smaller(int x){
        int ans = 0;
        for(Node *p = root; p != null;){
            if (x > p->v) ans += p->ch[0]->s + p->w, p = p->ch[1];
            else if (x == p->v){
                ans += p->ch[0]->s; break;
            }
            else p = p->ch[0];
        }
        return ans;
    }
    int get_bigger(int x){
        int ans = 0;
        for(Node *p = root; p != null;){
            if (x < p->v) ans += p->ch[1]->s + p->w, p = p->ch[0];
            else if (x == p->v){
                ans += p->ch[1]->s; break;
            }
            else p = p->ch[1];
        }
        return ans;
    }
}treap[maxn<<2];
inline int query(int l,int r,int x,bool type){
    if (l > r) return 0;
    int ret = 0; l += M; r += M;
    if (l == r) return type ? treap[l].get_bigger(x) : treap[l].get_smaller(x);
    if (!type) ret += treap[l].get_smaller(x) + treap[r].get_smaller(x);
    else ret += treap[l].get_bigger(x) + treap[r].get_bigger(x);
    if (!type){
        for(; l^r^1; l>>=1, r>>=1){
            if (!(l&1)) ret += treap[l^1].get_smaller(x);
            if (r&1) ret += treap[r^1].get_smaller(x);
        }
    }
    else{
        for(; l^r^1; l>>=1, r>>=1){
            if (!(l&1)) ret += treap[l^1].get_bigger(x);
            if (r&1) ret += treap[r^1].get_bigger(x);
        }
    }
    return ret;
}
int main(){
    n = read(); for(M = 1; M < n; M <<= 1); M--;
    rep(i,1,n){
        h[i] = read();
        for(int j=i+M; j>1; j>>=1)
        if (j&1) ans += treap[j^1].get_bigger(h[i]);
        for(int j=i+M; j; j>>=1)
        treap[j].ins(h[i],treap[j].root);
    }
    printf("%d
",ans);
    m = read();
    rep(i,1,m){
        l = read(); r = read();
        if (l > r) swap(l,r);
        if (l == r){
            printf("%d
",ans); continue;
        }
        ans += query(l+1,r-1,h[l],1) + query(l+1,r-1,h[r],0);
        ans -= query(l+1,r-1,h[l],0) + query(l+1,r-1,h[r],1);
        for(int j=l+M; j; j>>=1) treap[j].ins(h[r],treap[j].root);
        for(int j=l+M; j; j>>=1) treap[j].del(h[l],treap[j].root);
        for(int j=r+M; j; j>>=1) treap[j].ins(h[l],treap[j].root);
        for(int j=r+M; j; j>>=1) treap[j].del(h[r],treap[j].root);
        if (h[l] < h[r]) ans++; else if (h[l] > h[r]) ans--;
        swap(h[l],h[r]);
        printf("%d
",ans);
    }
    return 0;
}