给定一个长度为n的整数数列,请你计算数列中的逆序对的数量。
逆序对的定义如下:对于数列的第 i 个和第 j 个元素,如果满足 i < j 且 a[i] > a[j],则其为一个逆序对;否则不是。
输入格式:
第一行包含整数n,表示数列的长度;第二行包含 n 个整数,表示整个数列。(1≤n≤100000)
输出格式:
输出一个整数,表示逆序对的个数。
输入样例:
6
2 3 4 5 6 1
结尾无空行
输出样例:
5
结尾无空行
【算法思路】
按照归并的思路,将数组分为俩个部分:arr[l,mid] arr[mid+1,r]
且俩部分各自有序的情况下
逆序对分为三种状态:1,俩个数在左侧;2,俩个数在右侧;3,大的数在左侧,小的数在右侧。
假设归并已经求出来一二俩种情况的逆序对个数为merge(l,mid)+merge(mid+1,r)个。而我们只需要考虑第三种状态的求法。
在左右部分同时各自有序的情况下:如果右部分的当前数小于左部分的第i个数,那么这个右部分的当前数会依旧大于左部分的第i+1,i+2,i+3…mid个数。这些数一共有mid-i+1个!
#include<bits/stdc++.h>
using namespace std;
const int N = 1e6 + 10;
typedef long long ll;
int q[N], a[N];
ll merge_sort(int l, int r)
{
if (l == r) return 0;
int mid = l + r >> 1; //l+r右移一位,相当于除2
ll ans = merge_sort(l, mid) + merge_sort(mid + 1, r);
int i = l, j = mid + 1, k = 0;
while (i <= mid && j <= r) {
if (q[i] <= q[j])
a[k++] = q[i++];
else {
a[k++] = q[j++];
ans += mid - i + 1; //第一组中比j指引数大的有mid-i+1个
}
}
while (i <= mid) a[k++] = q[i++];
while (j <= r) a[k++] = q[j++];
for (int i = l, j = 0; i <= r; i++, j++)
q[i] = a[j];
return ans;
}
int main()
{
int n;
cin >> n;
for (int i = 0; i < n; i++)
cin >> q[i];
cout << merge_sort(0, n - 1) << endl;
return 0;
}