给定一个长度为 n
的整数数列,请你计算数列中的逆序对的数量。
逆序对的定义如下:对于数列的第 i
个和第 j 个元素,如果满足 i<j 且 a[i]>a[j]
,则其为一个逆序对;否则不是。
输入格式
第一行包含整数 n
,表示数列的长度。
第二行包含 n
个整数,表示整个数列。
输出格式
输出一个整数,表示逆序对的个数。
数据范围
1≤n≤100000
,
数列中的元素的取值范围 [1,109]
。
输入样例:
6
2 3 4 5 6 1
输出样例:
5
先来看一个错误的(正常人)------超时 复杂度 O(n^2)
#include<bits/stdc++.h>
using namespace std;
int main()
{
int n;
cin>>n;
int a[n];
int sum=0;
for(int i=0; i<n; i++){
scanf("%d",&a[i]);
}
for(int i=0; i<n; i++){
for(int j=0; j<n; j++){
if(i<j&&a[i]>a[j]) sum++;
}
}
cout<<sum<<endl;
return 0;
}
正确思路:分治 归并排序复杂度 O(n)
1.归并 把整个区间[l,r]分成[l,mid]和[mid+1,r]
2.递归排序两个子区间
3.归并两个有序数列为一个序列
过程:
第三种就是枚举后面部分里面的每个数,从前面的部分找有没有比它大的,记录,求s1+s2+...+sm.
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 1e5 + 10;
int a[N], tmp[N];
LL merge_sort(int q[], int l, int r)
{
if (l >= r) return 0;
int mid = l + r >> 1;//就是除以2
LL 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
{
res += mid - i + 1;//(mid - i + 1)这是一个si的和,因为归并排序后大小是排好的
tmp[k ++ ] = q[j ++ ];
}
//扫尾
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()
{
int n;
scanf("%d", &n);
for (int i = 0; i < n; i ++ ) scanf("%d", &a[i]);
cout << merge_sort(a, 0, n - 1) << endl;
return 0;
}