HDU 1394 Minimum Inversion Number（线段树/归并排序求逆序对数）

题目链接：
HDU 1394 Minimum Inversion Number
题意：
给出一组数，可以将每个数组首的数依次移到数组尾，求整个过程中（包括初始状态）的最少逆序对数。
分析：

• 用线段树求初始数组的逆序对数。对每个读入的数在线段树中查询之前有多少比它大的数字出现，那么这就是该数的逆序对数，累加就是整个初始数列的逆序对数。然后将读入的数插入到线段树叶子结点中,(指该数字已经出现了)，线段树中需要记录区间已经出现数字个数。对于每个从数列首移到数列尾的移动，可以用$tmp+=((n-1-a[i])-a[i])$；表示。对于每一个后移的$a[i]$会减少a[i]个逆序对(数据从0开始)，同时增加$n-a[i]-1$个逆序对，因为比$a[i]$大的数字个数是$n-a[i]-1$个，过程取最小值即可。

• 也可以用归并排序做。

  时间复杂度：$O(nlogn)$。

//1484K 62MS
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <climits>
#define lson(x) (x<<1)
#define rson(x) ((x<<1)|1)
using namespace std;
const int maxn=5050;

int n,ans;
int data[maxn];

struct SegTree{
    int left,right,num;//num是区间含有数字个数
}segtree[maxn<<2];

inline void build(int left,int right,int cur)
{
    segtree[cur].left=left;
    segtree[cur].right=right;
    segtree[cur].num=0;
    if(left==right) return ;
    int mid=(left+right)>>1;
    build(left,mid,lson(cur));
    build(mid+1,right,rson(cur));
}

inline void query(int a,int b,int cur)
{
    int left=segtree[cur].left;
    int right=segtree[cur].right;
    if(left==a&&right==b){
        ans+=segtree[cur].num;
        return; 
    }
    int mid=(left+right)>>1;
    if(b<=mid) query(a,b,lson(cur));
    else if(a>mid) query(a,b,rson(cur));
    else {
        query(a,mid,lson(cur));
        query(mid+1,b,rson(cur));
    }
}

inline void update(int cur,int pos)//将叶子结点pos的num赋为1，表示pos已经出现了
{
    int left=segtree[cur].left;
    int right=segtree[cur].right;
    if(left==right){//即：left=right=pos
        segtree[cur].num=1;
        return ;
    }
    int mid=(left+right)>>1;
    if(pos<=mid) update(lson(cur),pos);
    else update(rson(cur),pos);
    segtree[cur].num=segtree[lson(cur)].num+segtree[rson(cur)].num;
}

int main()
{
    //freopen("hdu1394in.txt","r",stdin);
    while(~scanf("%d",&n)){
        ans=0;
        build(0,n-1,1);
        for(int i=0;i<n;i++){
            scanf("%d",&data[i]);
            query(data[i],n-1,1);
            update(1,data[i]);
        }
        int tmp=ans;
        for(int i=0;i<n;i++){
            tmp+=((n-data[i]-1)-data[i]);//对于每一个后移的a[i]会减少a[i]个逆序对(数据从0开始)
            //同时增加n-a[i]-1个逆序对，因为比a[i]大的数字个数是n-a[i]-1个
            ans=min(ans,tmp);
        }
        printf("%d\n",ans);
    }
    return 0;
}

//1440K 62MS
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=5050;

int n,ans;
int data[maxn],tmp[maxn],T[maxn];

void Merge_sort(int* tmp,int low,int high,int* T)
{
    if(low+1<high){
        int mid=(low+high)>>1;
        Merge_sort(tmp,low,mid,T);
        Merge_sort(tmp,mid,high,T);
        int p=low,q=mid,i=low;
        while(p<mid||q<high){
            if(q>=high||(p<mid&&tmp[p]<tmp[q])){
                T[i++]=tmp[p++];
            }else {
                T[i++]=tmp[q++];
                ans+=(mid-p);
            }
        }
        for(int i=low;i<high;i++){
            tmp[i]=T[i];
        }
    }
}

int main()
{
    freopen("hdu1394in.txt","r",stdin);
    while(~scanf("%d",&n)){
        ans=0;
        for(int i=0;i<n;i++){
            scanf("%d",&data[i]);
            tmp[i]=data[i];   
        }
        Merge_sort(tmp,0,n,T);
        int res=ans;
        for(int i=0;i<n;i++){
            res+=(n-1-data[i]*2);
            ans=min(ans,res);
        }
        printf("%d\n",ans);
    }
    return 0;
}