【洛谷-分治】P1908 逆序对

题目描述

猫猫 TOM 和小老鼠 JERRY 最近又较量上了，但是毕竟都是成年人，他们已经不喜欢再玩那种你追我赶的游戏，现在他们喜欢玩统计。
最近，TOM 老猫查阅到一个人类称之为“逆序对”的东西，这东西是这样定义的：对于给定的一段正整数序列，逆序对就是序列中 $a_i>a_j$ 且 $i<j$ 的有序对。知道这概念后，他们就比赛谁先算出给定的一段正整数序列中逆序对的数目。注意序列中可能有重复数字。
Update:数据已加强。

输入输出格式

输入格式

第一行，一个数 $n$，表示序列中有 $n$个数。
第二行 $n$ 个数，表示给定的序列。序列中每个数字不超过 $10^9$。

输出格式

输出序列中逆序对的数目。

输入输出样例

输入样例 #1

6
5 4 2 6 3 1

输出样例 #1

11

说明

对于 $25\%$ 的数据，$n\le 2500$
对于 $50\%$ 的数据，$n\le 4\times 10^4$。
对于所有数据，$n\le 5\times 10^5$
请使用较快的输入输出
应该不会 $O(n^2)$ 过 50 万吧 by chen_zhe

解题思路

这题可以使用归并排序来做，每次合并时，如果右边小于左边的数，那么左边所有还没合并的数就都比该数大，也就是说左边所有还没合并的数和刚合并的右边的该数都为逆序对，每次合并时重复做该件事，那么最后统计出有多少逆序对了。
代码上，只需要在归并排序的合并代码上加一条统计sum的语句即可。

	while (li <= mid && ri <= r) {
		if (arr[li] <= arr[ri])tmp[ti++] = arr[li++];
		else tmp[ti++] = arr[ri++], sum += (long long)mid+1-li;
	}

完整代码

#include<iostream>

using namespace std;


int tmp[500005],nums[500005];
long long sum;
void margeSort(int arr[],int l,int r) {
	if (l >= r)return;
	int mid = l + ((r - l) >> 1);
	margeSort(arr, l, mid);
	margeSort(arr, mid + 1, r);
	int li = l,ri = mid+1,ti = 0;
	//每次右边(mid+1~r)放入时说明该数比arr[li]小，所以没放入的左边(li~mid)都比该数大,就添加(mid+1)-li个逆序对即可.
	while (li <= mid && ri <= r) {
		if (arr[li] <= arr[ri])tmp[ti++] = arr[li++];
		else tmp[ti++] = arr[ri++], sum += (long long)mid+1-li;
	}
	while (li <= mid)tmp[ti++] = arr[li++];
	while (ri <= r)tmp[ti++] = arr[ri++];
	for (int i = l; i <= r; i++)arr[i] = tmp[i - l];
}
int main() {
	int n;
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin >> n;
	for (int i = 0; i < n; i++)cin >> nums[i];
	margeSort(nums, 0, n-1);
	cout << sum;

	return 0;
}