对于一个包含N个非负整数的数组A[1..n],如果有i < j,且A[ i ]>A[ j ],则称(A[ i] ,A[ j] )为数组A中的一个逆序对。
逆序对数(逆序对的个数):给定一个数组,求该数组中包含多少个逆序对。
例如,数组(3,1,4,5,2)的逆序对有(3,1),(3,2),(4,2),(5,2),共4个。
最简单的方法是暴力枚举,时间复杂度O(n^2)。
/*/
题目:逆序对数 (暴力枚举)
时间:2015-04-14 11:11
作者:Miki
*//
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
int a[1000000+10],n,cnt;
int main()
{
// freopen("in.txt","r",stdin);
scanf("%d",&n);
for(int i=0;i<n;i++)
scanf("%d",&a[i]);
cnt=0;
for(int i=0;i<n;i++)
{
for(int j=i+1;j<n;j++)
{
if(a[i]>a[j])
cnt++;
}
}
printf("%d\n",cnt);
return 0;
}
当n很大时,枚举容易超时。下面介绍归并排序和树状数组计算逆序对数。
以POJ2299为例。
题意:
给出长度为n的序列,每次只能交换相邻的两个元素,问至少要交换几次才使得该序列为递增序列。
/*/
题目:逆序对数 (归并排序实现)
时间:2015-04-14 11:37
作者:Miki
*//
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int MAX=1000000+10;
int a[MAX],t[MAX],n;
long long int cnt;
void merge_sort(int A[],int x,int y,int T[])
{
if(y-x>1)
{
int m=x+(y-x)/2;
int p=x,q=m,i=x;
merge_sort(A,x,m,T);
merge_sort(A,m,y,T);
while(p<m||q<y)
{
if(q>=y||(p<m&&A[p]<=A[q]))
{
T[i++]=A[p++];
}
else
{
T[i++]=A[q++];
cnt+=m-p;//累加
}
}
for(int i=x;i<y;i++)
A[i]=T[i];
}
}
int main()
{
// freopen("in.txt","r",stdin);
while(scanf("%d",&n)!=EOF&&n)
{
for(int i=0;i<n;i++)
scanf("%d",&a[i]);
cnt=0;
merge_sort(a,0,n,t);
printf("%lld\n",cnt);
}
return 0;
}
最后是用树状数组实现。
//题目:逆序对数 (树状数组实现)
//时间:2015-04-14 20:24
//作者:Miki
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
typedef long long LL;
const int MAX=500000+10;
int bit[MAX],arr[MAX],n;
struct node
{
int val,index;
}a[MAX];
int sum(int i)
{
int s=0;
while(i>0)
{
s+=bit[i];
i-=i&(-i);
}
return s;
}
void add(int i,int x)
{
while(i<=n)
{
bit[i]+=x;
i+=i&(-i);
}
}
bool cmp(node x,node y)
{
return x.val<y.val;
}
int main()
{
// freopen("in.txt","r",stdin);
while(scanf("%d",&n)!=EOF&&n)
{
memset(bit,0,sizeof(bit));
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i].val);
a[i].index=i;
}
sort(a+1,a+n+1,cmp);
for(int i=1;i<=n;i++)
{
a[a[i].index].val=i;
}
LL ans=0;
for(int i=1;i<=n;i++)
{
add(a[i].val,1);
ans+=i-sum(a[i].val);
}
printf("%lld\n",ans);
}
return 0;
}