归并排序——逆序对数量

题目描述

给定一个长度为 n 的整数数列，请你计算数列中的逆序对的数量。

逆序对的定义如下：对于数列的第 i 个和第 j 个元素，如果满足 i<j 且 a[i]>a[j]，则其为一个逆序对；否则不是。

输入格式

第一行包含整数 n，表示数列的长度。

第二行包含 n 个整数，表示整个数列。

输出格式

输出一个整数，表示逆序对的个数。

数据范围

1≤n≤100000，
数列中的元素的取值范围 [1,1e9]。

输入样例

6
2 3 4 5 6 1

输出样例

5

注释版代码

此题参考归并排序——模板题-CSDN博客

#include<iostream>
using namespace std;
typedef long long LL;//由于数据范围较大，所以使用long long
const int N=100010;
int q[N],tmp[N];
//逆序对就是两个数前面的数大于后面的数，我们使用归并排序的话，有三种情况：
//第一种是两个数都在左边部分(l，mid)，第二种是两个数都在右边部分(mid+1,r)
//还有一种情况就是前面的数在左半部分，后面的数在右半部分
//所以逆序对的数量=左半部分加上右半部分加上两边的
LL merge_sort(int l,int r)
{
	if(l>=r)
	return 0;
	int mid=l+r>>1;
	LL res=merge_sort(l,mid)+merge_sort(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++];
			res+=mid-i+1;//这里就是第三种情况
			//当q[i]>q[j]时，那么左半部分i后面的所有数都是大于j的，那么这个j的逆序对个数就是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 res;
}
int main()
{
	int n;
	scanf("%d",&n);
	for(int i=0;i<n;i++)
	{
		scanf("%d",&q[i]);
	}
	printf("%d",merge_sort(0,n-1));
	return 0;
}