题目(7.6)链接: The Number of Inversions
知识点:排序,分治。
技巧:套用归并排序模板,使得复杂度变成了O(Nlog(N)),远小于采用冒泡排序求逆序数时的O(N^2)。重点在与添加了cnt和v1、v2、v3。
#include <iostream>
#include <cstdio>
using namespace std;
typedef long long LL;
const int Max = 200005;
const int M = 1000000005;
int A[Max];
int a[Max/2], b[Max/2];
LL Merge(int S[], int left, int right, int mid)//归并(归并时比较排序)
{
int n1 = mid-left, n2 = right-mid;//以mid为界限,分割为两个数组
LL cnt = 0;
for(int i=0; i<n1; i++)
{
a[i] = S[i+left];
}
for(int i=0; i<n2; i++)
{
b[i] = S[i+mid];
}
a[n1] = M, b[n2] = M;//确保最后一个数较大,利于以下排序(M超出了S中元素的范围)
int k=left, ka=0, kb=0;
//Count++;
while(k<right)
{
if(a[ka]>b[kb])//升序排列,一定为大于,保证了稳定排序
{
S[k++] = b[kb++];
cnt += n1-ka;///重点:归并时,当排入后一个数组的元素时,查看第一个数组还有多少元素会排它的后面
}
else
{
S[k++] = a[ka++];
}
}
return cnt;
}
LL MergeSort(int S[], int left, int right)//用于分割数组,分别排序(分治)
{
LL v1, v2, v3;
if(left+1<right)//判断是否可以分割
{
int mid = (left+right)/2;
v1 = MergeSort(S, left, mid);
v2 = MergeSort(S, mid, right);
v3 = Merge(S, left, right, mid);//归并
return v1+v2+v3;
}
else
return 0;
}
int main()
{
int n;
scanf("%d", &n);
for(int i=0; i<n; i++)
{
scanf("%d", &A[i]);
}
LL ans = MergeSort(A, 0, n);
printf("%lld\n", ans);
return 0;
}