【洛谷-分治】P1908 逆序对

最新推荐文章于 2023-01-10 15:33:58 发布

原创最新推荐文章于 2023-01-10 15:33:58 发布 · 361 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#分治算法 #算法 #排序算法

算法题专栏收录该内容

15 篇文章

订阅专栏

本文介绍了如何利用归并排序算法快速计算给定正整数序列中的逆序对数量，通过合并过程中判断元素大小来统计逆序对，适用于数据规模达50万的高效解决方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

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

输入输出格式

输入格式

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

输出格式

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

输入输出样例

输入样例 #1

6
5 4 2 6 3 1

输出样例 #1

说明

对于 $25%25\%$ 的数据， $\leq 2500$
对于 $50%50\%$ 的数据， $\leq 4 \times 10^4$ 。
对于所有数据， $\leq 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;
}