给定一个长度为n的整数数列,请你计算数列中的逆序对的数量。
逆序对的定义如下:对于数列的第 i 个和第 j 个元素,如果满足 i < j 且 a[i] > a[j],则其为一个逆序对;否则不是。
输入格式
第一行包含整数n,表示数列的长度。
第二行包含 n 个整数,表示整个数列。
输出格式
输出一个整数,表示逆序对的个数。
数据范围
1≤n≤100000
输入样例:
6
2 3 4 5 6 1
输出样例:
5
思路
利用归并排序将可能的情况分成三种
①左半边数组逆序对的数量:merge_sort(l , mid)
②右半边数组逆序对的数量: merge_sort(mid + 1 , r )
③左边数组值比右边数组大的逆序对数量:mid -i + 1(因为半边数组已经是排好序的,如果q[i]>q[j],那么左半边数组I后面的数也和j形成逆序对)
#include <iostream>
using namespace std;
const int N = 100010;
typedef long long LL;
int q[N],tmp[N];
int n;
LL merge_sort(int q[], int l , int r)
{
if( l >= r) return 0;
int mid = l + r >>1;
LL cnt = 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++ ];
cnt += mid - i + 1;
}
}
while(i <= mid) tmp[k++ ] = q[i++];
while(j <= r) tmp[k++ ] = q[j++];
for(int i = l , j = 0 ; i <= r ; i++ ,j++) q[i] = tmp[j];
return cnt;
}
int main()
{
cin>>n;
for(int i = 0 ; i < n ; i++)
cin>>q[i];
cout<<merge_sort(q , 0 , n-1)<<endl;
return 0;
}