给定一个长度为 n 的整数数列,请你计算数列中的逆序对的数量。
逆序对的定义如下:对于数列的第 i 个和第 j个元素,如果满足 i<j 且 a[i]>a[j],则其为一个逆序对;否则不是。
输入格式
第一行包含整数 n,表示数列的长度。
第二行包含 n个整数,表示整个数列。
输出格式
输出一个整数,表示逆序对的个数。
数据范围
1≤n≤1000001≤�≤100000,
数列中的元素的取值范围 [1,109][1,109]。
输入样例:
6
2 3 4 5 6 1
输出样例:
5
代码如下:
#include<iostream>
using namespace std;
const int N=1000010;
int n;
int q[N],tmp[N];
long long res=0;
long long merge_sort (int q[],int l,int r){
if(l>=r){
return 0;
}
int mid=l+r>>1;
res=merge_sort(q,l,mid)+merge_sort(q,mid+1,r);
int k=0,i=l,j=mid+1;
while(i<=mid&&j<=r){
if(q[i]<=q[j]){
tmp[k++]=q[i++];
}else{
tmp[k++]=q[j++];
res = res+mid - i + 1;
}
}
while(i<=mid){
tmp[k++]=q[i++];
}
while(j<=r){
tmp[k++]=q[j++];
}
for(i=l,j=0;i<=r;i++,j++){
q[i]=tmp[j];
}
return res;
}
int main(){
scanf("%d",&n);
for(int i=0;i<n;i++){
scanf("%d",&q[i]);
}
long long s=merge_sort(q,0,n-1);
printf("%lld",s);
return 0;
}
当我们在合并两个已排序的数组时,我们需要统计逆序对的数量。逆序对是指左边数组中的一个元素大于右边数组中的一个元素。
在归并排序的过程中,当我们合并两个已排序的数组时,我们比较左边数组的元素和右边数组的元素。如果左边数组的元素大于右边数组的元素,那么左边数组中这个元素以及它后面的所有元素都与右边数组的这个元素构成逆序对。
现在想象一下,当我们在合并这两个已排序的数组时,如果我们发现左边数组的某个元素比右边数组的某个元素大,这意味着右边数组中的所有剩余元素都比左边数组中当前元素小,因为这两个数组都是已经排序好的。
所以,当我们找到一个左边数组中的元素大于右边数组中的元素时,我们可以知道在右边数组中这个元素后面的所有元素都和左边数组中的当前元素构成逆序对。
mid - i + 1
表示左边数组从当前位置 i
到中间位置 mid
的元素数量(包括 q[i]
和 q[mid]
),这个数量就是构成逆序对的个数。
res += mid - i + 1;
就是把当前这一次合并中发现的逆序对的数量加到 res
中,最终返回的 res
就是整个数组中的逆序对的数量。