给定一个长度为 n的整数数列,请你计算数列中的逆序对的数量。
逆序对的定义如下:对于数列的第 i 个和第 j个元素,如果满足 i<j且 a[i]>a[j],则其为一个逆序对;否则不是。
输入格式
第一行包含整数 n,表示数列的长度。
第二行包含 n个整数,表示整个数列。
输出格式
输出一个整数,表示逆序对的个数。
数据范围
1≤n≤100000,
数列中的元素的取值范围 [1,109]。
输入样例:
6
2 3 4 5 6 1
输出样例:
5
题目分析:
题目说求逆序对数量,那我们可以套用归并排序模板。主要分为三种情况,两个点都在mid 左边,右边,或者一个在左边,一个在右边,前面两种情况我们可以直接递归去求,然而对于最后一种情况其实就是mid - i + 1。
代码如下:
#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
const int N = 100010;
int t[N], a[N];
int n;
long long res;
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 i = l, j = mid + 1, k = 0;
while(i <= mid && j <= r)
{
if(q[i] <= q[j]) t[k ++] = q[i ++];
else {
res += mid - i + 1;
t[k ++] = q[j ++];
}
}
while(i <= mid) t[k ++] = q[i ++];
while(j <= r) t[k ++] = q[j ++];
for(int i = l, j = 0; i <= r; i ++, j ++) q[i] = t[j];
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0), cout.tie(0);
cin >> n;
for(int i = 0; i < n; i ++) cin >> a[i];
merge_sort(a, 0, n-1);
cout << res << endl;
return 0;
}