HDOJ1394 Minimum Inversion Number【线段树】

//往线段树中添加数据，每个结点记录的是
//当前结点范围已经插入的数字个数
//如果p点在左子树上，就累加右子树根节点上的记录
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define maxn 5001
int sum[maxn<<2],num[maxn+1];
int Min,s;
void build()
{
    memset(sum,0,sizeof(sum));
}
void insert(int p,int l,int r,int rt)
{
    int m;
    if(l<=p&&p<=r)
        sum[rt]++;
    if(l==r)
        return;
    m=((l+r)>>1);
    if(p<=m)
    {
        insert(p,lson);
        s+=sum[rt<<1|1];
    }        
    if(p>m)
        insert(p,rson);
}
int main()
{
    int n,i;
    while (scanf("%d",&n)!=EOF)
    {
        //初始化
        s=0;
        build();
        //统计，得到s值
        for (i=0;i<n;i++)
        {
            scanf("%d",&num[i]);
            insert(num[i]+1,1,n,1);
        }
        //求出Min
        Min=s;
        for (i=0;i<n;i++)
        {
            s=s-num[i]+n-1-num[i];
            Min=Min<s?Min:s;
        }
        printf("%d\n",Min);
    }
    return 0;
}