poj 2299 Ultra-QuickSort

题目链接如下所示：

2299 -- Ultra-QuickSort

这题其实就是转化成求逆序数的问题，比如题目里边的例子，9 1 0 5 4，那么我们最快的做法就是先将最大的数字移动到最后面，那么其他数字在交换的时候就不必考虑这个数字了，因为他的位置已经固定住了。

这里要注意的问题就是，要压缩数据，比如上面的9 1 0 5 4压缩成5 2 1 4 3，这样不必开太大的空间，除此之外题目规模是500000，那么最坏情况的逆序对数应该是10^11级别，得开个long long的结果。

代码如下所示：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<ctime>
#include<vector>
#include<algorithm>
#include<cstring>
#include<set>
#include<map>
#include<queue>
#include<string>
#include<cmath>
#include<climits>

using namespace std;
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1

const int MAXN=500001;
int tree[MAXN<<2];
int num[MAXN];
int sort_num[MAXN];
long long ans;

void push_up(int rt){
    tree[rt]=tree[rt<<1]+tree[rt<<1|1];
}

void build(int l,int r,int rt){
    tree[rt]=0;
    if (l==r){
        return;
    }
    int mid=(l+r)>>1;
    build(lson);
    build(rson);
}

void update(int a,int l,int r,int rt){
    if (l==r && l==a){
        tree[rt]++;
        return;
    }
    int mid=(l+r)>>1;
    if (a<=mid){
        update(a,lson);
    }else{
        update(a,rson);
    }
    push_up(rt);
}

int query(int a,int b,int l,int r,int rt){
    if (a<=l && b>=r){// 发现逆序对
        return tree[rt];
    }
    int res=0;
//    cout<<l<<","<<r<<endl;
    int mid=(l+r)>>1;
    if (a<=mid){
        res+=query(a,b,lson);
    }
    if (b>mid){
        res+=query(a,b,rson);
    }
    return res;
}

int main(){
    int n,a;
    while (scanf("%d",&n)){
        if (n==0) break;
        ans=0;
        build(1,n,1);
        for (int i = 1; i <= n; ++i) {
            scanf("%d",&num[i]);
            sort_num[i]=num[i];
//            cout<<num[i]<<endl;
        }
        sort(sort_num+1,sort_num+1+n);
        for (int i = 1; i <= n; ++i) {
            a= lower_bound(sort_num+1,sort_num+1+n,num[i])-sort_num;
//            cout<<"num:"<<num[i]<<",rank:"<<a<<endl;
            ans+=query(a,n,1,n,1);
            update(a,1,n,1);
        }
        printf("%lld\n",ans);
    }
    return 0;
}