线段树 Minimum Inversion Number D题

题目：Minimum Inversion Number
题意：有0～n-1数字组成的序列,然后进行这样的操作,每次将最前面一个元素放到最后面去会得到一个序列,每得到一个序列都可得出该序列的逆序数（如果一对数的前后位置与大小顺序相反，即前面的数大于后面的数，那么它们就称为一个逆序。一个排列中逆序的总数就称为这个排列的逆序数）。要求求出最小的逆序数。
思路：线段树维护的是在此区间内出现的数字的个数。如：数字3出现后，则区间包含3的都加1；
以给出的样例为例：
10
1 3 6 9 0 8 5 7 4 2
第一个数是1，则tree[10]+=1,tree[8]+=1,tree[4]+=1…一直向上更新至tree[1]，更新完毕后可计算数字1前面出现了几个比1 大的数字=（数字1出现的位置—1）—已经输入的小于1的数字;
第二个数是3，则tree[9]+=1,tree[5]+=1…一直到tree[1]，更新完毕后可计算数字3前面出现了几个比3 大的数字=（数字3出现的位置—1）—已经输入的小于3的数字的个数，此时为：num=2（第二个数）—1—1（前面已经输入了一个小于3的数字）；
…
第十个数是2，num=10-1-2（两个小于2的数：0，1）；
把每个num相加，得到第一个序列的逆序数n1。
然后，计算把第一个数放到序列最后所得的新序列的逆序数。
会发现新的逆序数n2=n1-（n-1）+(N-n)(N是序列总长度，n是放到序列最后的那个数字）
这样就可以for语句循环比较最小的逆序数。
代码如下：

#include <iostream>
#include <stdio.h>
#define MAXN 5005

using namespace std;

int tree[MAXN<<2];
int N,nxs[MAXN];//nxs存的是第i个出现的数字，如上例nxs[1]=1,nxs[2]=3;

void build(int k,int l,int r)
{
    if(l==r)
    {
        tree[k]=0;
        return ;
    }
    int mid=(l+r)/2;
    build(k*2,l,mid);
    build(k*2+1,mid+1,r);
    tree[k]=tree[k*2]+tree[k*2+1];
}

void updata(int k,int l,int r,int x,int p)
{
    if(l==r)
    {
        tree[k]=p;
        return ;
    }
    int mid=(l+r)/2;
    if(x<=mid)
    {
        updata(k*2,l,mid,x,p);
    }
    else
    {
        updata(k*2+1,mid+1,r,x,p);
    }
    tree[k]=tree[k*2]+tree[k*2+1];
}

int sum(int k,int l,int r,int x,int y)
{
    if(x<=l&&y>=r)
        return tree[k];
    int mid=(l+r)/2;
    int ret=0;
    if(x<=mid)
    {
        ret+=sum(k*2,l,mid,x,y);
    }
    if(y>mid)
    {
        ret+=sum(k*2+1,mid+1,r,x,y);
    }
    return ret;
}

int minn(int a,int b)
{
    if(a<b)
        return a;
    else
        return b;
}

int main()
{
    while(scanf("%d",&N)!=EOF){
    build(1,1,N);
    int n;
    int num=0,ans;
    for(int i=1;i<=N;i++)
    {
        scanf("%d",&n);
        n=n+1;
        nxs[i]=n;
        updata(1,1,N,n,1);
        num+=i-1-(sum(1,1,N,1,n)-1);//计算逆序数
    }
    ans=num;
    for(int i=1;i<N;i++)
    {
        num=num-(nxs[i]-1)+(N-nxs[i]);//计算下一个序列的逆序数，公式看上面
        ans=minn(num,ans);
    }
    printf("%d\n",ans);
    }
    return 0;
}