Minimum Inversion Number 1394(线段树法)

http://acm.hdu.edu.cn/showproblem.php?pid=1394

线段树记录区间[a,b]里面出现了多少个数字，查询时只要查询区间[x,n-1]有多少个数出现过，就知道前面有多少个数比x大，也就知道x有多少逆序数。

把一个序列最前面的数字x移到最后就会产生n-1-x-x个逆序：

x在最前时它后面有x个数(0到x-1)比x小，所以把x移到最后以后逆序数要减x个;

x在最前时它后面有n-1-x个数(x+1到n-1)比x大，所以把x移到最后以后逆序数增加n-1-x个;

所以，最前面的数字x移到最后就会新产生n-2*x-1个逆序

#include <iostream>
#include <cstdio>
#include <cstring>
#define lson l,mid,i<<1
#define rson mid+1,r,i<<1|1
using namespace std;
int n;
int sum[5010*3];
int query(int ql,int qr=n-1,int l=0,int r=n-1,int i=1)
{
    if(ql<=l&&r<=qr) return sum[i];
    int ret=0,mid;
    mid=(l+r)>>1;
    if(ql <= mid) ret+=query(ql,qr,lson);
    if(qr > mid) ret+=query(ql,qr,rson);
    return ret;
}
void update(int ql,int l=0,int r=n-1,int i=1)
{
    if(l==r) {sum[i]++;return;}
    int mid=(l+r)>>1;
    if(ql<=mid) update(ql,lson);
    else update(ql,rson);
    sum[i]=sum[i<<1]+sum[i<<1|1];
}
int x[5000];
int main()
{
    while(cin>>n)
    {
        int ret=0;
        memset(sum,0,sizeof(sum));//这里相当于建树
        for(int i=0;i<n;i++)
        {
            scanf("%d",x+i);
            ret+=query(x[i]);
            update(x[i]);
        }
        int ans=ret;
        for(int i=0;i<n;i++)
        {
            ret+=n-x[i]-x[i]-1;
            ans=min(ans,ret);
        }
        printf("%d\n",ans);
    }
    return 0;
}