题目如下:
给定一个长度为 n 的整数数列,请你计算数列中的逆序对的数量。
逆序对的定义如下:对于数列的第 i个和第 j个元素,如果满足 i<j且 a[i]>a[j],则其为一个逆序对;否则不是。
输入格式
第一行包含整数 n,表示数列的长度。
第二行包含 n 个整数,表示整个数列。
输出格式
输出一个整数,表示逆序对的个数。
数据范围
1≤n≤100000,
数列中的元素的取值范围 [1,1000000000]。
输入样例:
6
2 3 4 5 6 1
输出样例:
5
题解:
对于数列的第 i 个和第 j 个元素,如果满足 i < j 且 a[i] > a[j],则其为一个逆序对。
重要的地方在于,一个元素可以不只是在一个逆序对中存在。如果 k > j > i 且 a[i] > a[j] > a[k],那么这里有两个含 a[i] 的逆序对,分别是 (a[i], a[j]) 和 (a[i], a[k]), a[i]是可以使用多次的。
第二步是分析问题,这里我们可以使用分治法解决问题。
我们将序列从中间分开,将逆序对分成三类:
两个元素都在左边;
两个元素都在右边;
两个元素一个在左一个在右。
因此这就是我们算法的大致框架:
1、假定归并排序把序列进行排序并返回逆序对的数量;
2、当两个数在左边或者右边的时候就是直接调用归并排序返回;
3、当出现一个在左,一个在右的时候,就是第三种情况,利用多路归并算法进行求解;
4、前面两种由于递归的性质最后都会变成第三种情况。
这个时候我们注意到一个很重要的性质,左右半边的元素在各自任意调换顺序,不影响第三步计数,因此我们可以数完就给它排序。这么做的好处在于,如果序列有序,会让第三步计数很容易。
如果无序暴力数的话这一步是O(n^2)。
比如序列是这样:
4 5 6 | 1 2 3
4 5 6
1 2 3
当你发现 4 比 3 大的时候,也就是说右边最大的元素都小于左边最小的元素,那么左边剩下的5和6都必然比右边的所有元素大,因此就可以不用数5和6的情形了,直接分别加上右半边的元素个数就可以了,这一步就降低到了O(n), 我们知道递归式 T(n) = 2T(n/2)+O(n) = O(nlogn)的,所以排序的成本是可以接受的,并且这一问题下,可以很自然地使用归并排序。
代码如下:
#include<cstdio>
using namespace std;
const int N=1e6+10;
int n,q[N],tmp[N];
long long ans=0; //根据题目数据要求,避免数据溢出错误
void merge_sort(int q[],int l,int r) //归并排序
{
if(l>=r) return;
int mid=l+r>>1;
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++];
ans+=(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];
}
int main()
{
scanf("%d",&n);
for(int i=0;i<n;i++)
scanf("%d",&q[i]);
merge_sort(q,0,n-1);
printf("%lld",ans); //万恶的C语言,必须使用"%lld",否则数据溢出
return 0;
}
参考文献:
SOLORED 的题解及评论区大佬
y总算法基础课