【洛谷】P1908 逆序对（归并排序+快速读入）题解

原题链接：https://www.luogu.org/problem/P1908

题目描述

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


输入输出格式

输入格式：

第一行，一个数n，表示序列中有n个数。

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

输出格式：

给定序列中逆序对的数目。


输入输出样例

输入样例#1：

6
5 4 2 6 3 1

输出样例#1：

11


说明

时空限制：1000ms 125M

对于25%的数据，n ≤ 2500

对于50%的数据，n ≤ 4×10^4

对于所有数据，n ≤ 5×10^5

请使用较快的输入输出

应该不会n方过50万吧 by chen_zhe


思路：一道归并排序模板题，多了记录逆序对的步骤以及快速读入模板。

归并排序的思想就是二分+递归，不断二分，比较a[i]和a[j]的大小（a[i]属于左区间，a[j]属于右区间），若a[i]<a[j]，则将a[i]复制到b[k]中，i和k都加1，否则将a[j]复制到b[k]中，将j，k加1，最后再将数组b[ ]移动到数组a[ ]中，然后继续合并。同时，每次合并都要加上此次逆序对的数目。

代码如下：

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int N=1e6;
typedef long long ll;
int a[N],b[N],n;	//数组a[]表示输入的序数，数组b[]表示重排后的序数 
ll total=0;		//必须定义为long long型，记录逆序对的数量 
inline int read()	//快速读入模板
{
    char c=getchar();int num=0;
    for(;!isdigit(c);c=getchar());
    for(;isdigit(c);c=getchar())
        num=num*10+c-'0';
    return num;
}
void mergesort(int l,int r){	//归并排序模板，只是多了对逆序对的记录 
	if(l==r)
		return;
	int mid=(l+r)/2;
	mergesort(l,mid);	//不断递归二分 
	mergesort(mid+1,r);
	int x=l,y=mid+1,z=l;
	while(x<=mid&&y<=r){
		if(a[x]>a[y]){
			b[z++]=a[y++];	//先赋值后加1右移 
			total=total+mid-x+1;
		}
		else
			b[z++]=a[x++];
	}
	while(x<=mid)
		b[z++]=a[x++];
	while(y<=r)
		b[z++]=a[y++];
	for(int i=l;i<=r;i++)	//把排序后的重新复制回a数组 
		a[i]=b[i];
}
int main() {
	cin>>n;
	for(int i=1;i<=n;i++)
		a[i]=read();
	mergesort(1,n);
	cout<<total<<endl;
	return 0;
}