求逆序数
时间限制:2000 ms | 内存限制:65535 KB
难度:5
描述
在一个排列中,如果一对数的前后位置与大小顺序相反,即前面的数大于后面的数,那么它们就称为一个逆序。一个排列中逆序的总数就称为这个排列的逆序数。
现在,给你一个N个元素的序列,请你判断出它的逆序数是多少。
比如 1 3 2 的逆序数就是1。
输入
第一行输入一个整数T表示测试数据的组数(1<=T<=5)
每组测试数据的每一行是一个整数N表示数列中共有N个元素(2〈=N〈=1000000)
随后的一行共有N个整数Ai(0<=Ai<1000000000),表示数列中的所有元素。
数据保证在多组测试数据中,多于10万个数的测试数据最多只有一组。
输出
输出该数列的逆序数
样例输入
2
2
1 1
3
1 3 2
样例输出
0
1
思路:利用归并排序,将排序的归并过程加一段代码,用于统计这个区间的逆序数。
#include<stdio.h>
#define N 1000010
int temp[N],arr[N];
long long ans;
void merge(int x,int mid, int y){
int i,j=mid+1,k;
for(i=x,j=mid+1,k=0;i<=mid&&j<=y;){
if(arr[i]<=arr[j]) {
temp[k++]=arr[i++];
} else {
// 计算左序列中,arr[i]以及之后的数(截止到mid),和arr[j]构成的逆序数
ans += (i - mid + 1);
temp[k++]=arr[j++];
}
}
while(i<=mid) temp[k++]=arr[i++];
while(j<=y) temp[k++]=arr[j++];
for(i=0;i<=y-x;i++){
arr[i+x]=temp[i];
}
}
void mergeSort(int i, int j){
int mid;
if(i<j){
mid=(i+j)/2;
mergeSort(i,mid);
mergeSort(mid+1,j);
merge(i,mid,j);
}
}
int main(){
int T,i,n;
scanf("%d",&T);
while(T--){
scanf("%d",&n);
for(i=0,ans=0;i<n;i++){
scanf("%d",&arr[i]);
}
mergeSort(0,n-1);
printf("%lld\n",ans);
}
return 0;
}