POJ2299 逆序数

传送门@百度。。

treap好久没写果然有点生疏了，注意答案是long long

#include<set>
#include<queue>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
using namespace std;
const int N = 500010;
#define Ch1 (T[i].s[0])
#define Ch2 (T[i].s[1])
#define For(i,n) for(int i=1;i<=n;i++)
#define Rep(i,l,r) for(int i=l;i<=r;i++)

void read(int &v){
    char ch=getchar();
    int num=0;
    while(ch>'9'||ch<'0') ch=getchar();
    while(ch>='0'&&ch<='9'){
        num=num*10+ch-'0';
        ch=getchar();
    }
    v=num;
}

struct treap{
    int v,size,pri;
    int s[2];
    void Sets(int x,int y){v=x;pri=y;size=1;}
}T[N];

int n,x,cnt,root;
long long ans;

void Update(int i){
    T[i].size=T[Ch1].size+T[Ch2].size+1;
}

void Rot(int &y,int f){
    int x=T[y].s[!f];
    T[y].s[!f]=T[x].s[f];
    T[x].s[f]=y;
    Update(y);Update(x);
    y=x;
}

void Insert(int &i,int x){
    if(!i){
        T[i=++cnt].Sets(x,rand());
        T[i].s[0]=T[i].s[1]=0;
        return;
    }
    int f=T[i].v>x;
    Insert(T[i].s[!f],x);
    if(T[T[i].s[!f]].pri>T[i].pri) Rot(i,f);
    else                           Update(i);    
}

int find(int i,int x){
    if(T[i].v==x) return T[Ch1].size+1;
    if(T[i].v<x) return T[Ch1].size+1+find(Ch2,x);
    if(T[i].v>x) return find(Ch1,x);
}

int main(){
    while(read(n),n){
        ans=0;
        cnt=root=0;
        For(i,n){
            read(x);
            Insert(root,x);     
            ans+=i-(find(root,x));         
        }
        cout<<ans<<endl;
    }
    return 0;
}