HDU 1394 Minimum Inversion Number(线段树:单点更新,区间求和)

HDU 1394 Minimum Inversion Number(线段树:单点更新,区间求和)

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

题意：

        有一个n个整数的排列，这n个整数就是0,1,2,3...n-1这n个数(但不一定按这个顺序给出)。现在先计算一下初始排列的逆序数，然后把第一个元素a1放到an后面去，形成新排列a2 a3 a4...an a1，然后再求这个排列的逆序数。继续执行类似操作(一共要执行n-1次)直到产生排列an a1 a2...an-1为止。计算上述所有排列的逆序数，输出最小逆序数。

分析:

        首先对于读入的每个a[i],执行a[i]++操作.使得序列变成1到n之间.

        首先利用线段树求出初始序列a1,a2…an的逆序数ans.类似于树状数组对逆序的求法.详解:

http://blog.csdn.net/u013480600/article/details/21293613.

        维护一棵线段树,树中的每个叶子节点表示的是这个节点是否出现过,比如叶子节点i控制的范围是[2,2],那么sum[i]=1表示2这个数已经出现了,如果sum[i]=0表2这个数还没出现.如果j节点控制范围[4,8],那么sum[j]=3表示[4,8]区间有3个数出现了.

        如果当前处理的是a[i],那么用query(a[i]+1,n,1,1,n)找到在a[i]之前出现的比a[i]值大的数有多少个,那么这个值就是a[i]的逆序数.总的逆序数ans=所有a[i]的逆序数之和.

        然后因为序列是一个n的排序(从1到n)  ,a2,a3,…an,a1的逆序数=ans-(a[1]-1 )+(n-a[1] )=ans+n+1-2*a[1]

        由此可得:如果操作前的逆序数是ans,当前要把a[i]放到最后面的话,那么操作后的逆序数是ans+n+1-2*a[i]

AC代码:46ms

#include<cstdio>#include<cstring>#include<algorithm>using namespace std;const int MAXN=5000+100;int sum[MAXN*4];int a[MAXN];#define lson i*2,l,m#define rson i*2+1,m+1,rvoid PushUp(int i){    sum[i]=sum[i*2]+sum[i*2+1];}void build(int i,int l,int r){    if(l==r)    {        sum[i]=0;        return ;    }    int m=(l+r)/2;    build(lson);    build(rson);    PushUp(i);}int query(int ql,int qr,int i,int l,int r){    if(ql<=l&&r<=qr)        return sum[i];    int m=(l+r)/2;    int res=0;    if(ql<=m)res += query(ql,qr,lson);    if(qr>m)res+=query(ql,qr,rson);    return res;}void update(int p,int i,int l,int r){    if(l==r)    {        sum[i]++;        return ;    }    int m=(l+r)/2;    if(p<=m)update(p,lson);    else update(p,rson);    PushUp(i);}int main(){    int n;    while(scanf("%d",&n)==1&&n)    {        int ans=0;        memset(sum,0,sizeof(sum));//这个操作代替了build(1,1,n);        for(int i=1;i<=n;i++)        {            scanf("%d",&a[i]);            a[i]++;            ans += query(a[i]+1,n,1,1,n);            update(a[i],1,1,n);        }        int min_num=ans;        for(int i=1;i<n;i++)        {            ans = ans+n+1-2*a[i];            min_num=min(min_num , ans);        }        printf("%d\n",min_num);    }    return 0;}