Minimum Inversion Number HDU 1394

用树状数组求逆序数。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define maxn 5004
int a[maxn];
int n=maxn;
int lowbit(int i)
{
return i&(-i);
}
int Sum(int i)
{
int sum=0;
while(i>=1)
{
sum+=a[i];
i-=lowbit(i);
}
return sum;
}
void Update(int i,int tem)
{
while(i<=n)
{
a[i]+=tem;
i+=lowbit(i);
}
}
int main()
{
int m,i;
int sum,min,k;
int b[maxn],c[maxn];
while(scanf("%d",&m)!=EOF)
{
memset(a,0,sizeof(a));
memset(b,0,sizeof(b));
memset(c,0,sizeof(c));
for(i=1;i<=m;i++)
{
scanf("%d",&b[i]);//用b接受原数据
b[i]++;//防止输入0，造成死循环。
}
sum=0;
for(i=1;i<=m;i++)
{
Update(b[i],1);//离散化数据
sum+=Sum(maxn)-Sum(b[i]);//求出比这个数小的
}
min=sum;
for(i=1;i<=m;i++)//把第一个数变换到最后.
//Sum(b[i]-1)为变换之前第一个数构成的逆序数量，(m-1-Sum(b[i]-1)为变换之后增加的逆序的数量。
{
k=sum-Sum(b[i]-1)+(m-1-Sum(b[i]-1));
if(k<min)
min=k;
sum=k;
} printf("%d\n",min);
}
return 0;
}