逆序对 洛谷 P1908 归并排序 or 树状数组

 题目描述

猫猫 TOM 和小老鼠 JERRY 最近又较量上了，但是毕竟都是成年人，他们已经不喜欢再玩那种你追我赶的游戏，现在他们喜欢玩统计。

最近，TOM 老猫查阅到一个人类称之为“逆序对”的东西，这东西是这样定义的：对于给定的一段正整数序列，逆序对就是序列中 a_i>a_jai​>aj​ 且 i<ji<j 的有序对。知道这概念后，他们就比赛谁先算出给定的一段正整数序列中逆序对的数目。注意序列中可能有重复数字。

Update:数据已加强。

输入格式

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

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

输出格式

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

输入输出样例

输入 #1复制

6
5 4 2 6 3 1

输出 #1复制

11

说明/提示

对于 25\%25% 的数据，n \leq 2500

对于 50\%50% 的数据，n \leq 4 \times 10^4n≤4×104。

对于所有数据，n \leq 5 \times 10^5n≤5×105

请使用较快的输入输出

应该不会 O(n^2)过 50 万吧 by chen_zhe

归并排序

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN = 500010;
int num[MAXN],temp[MAXN];
ll ans;
void msort(int b,int e)
{
	if(b == e) return;
	int mid = (b+e) >> 1,i = b,j = mid+1,k = b;
	msort(b,mid),msort(mid+1,e);
	while(i <= mid && j <= e){
		if(num[i] <= num[j]) temp[k++] = num[i++];
		else{
			temp[k++] = num[j++];
			ans += mid - i + 1;
		}
	}
	while(i <= mid) temp[k++] = num[i++];
	while(j <= e) temp[k++] = num[j++];
	for(int l = b;l <= e; l++)
		num[l] = temp[l];
 } 
int main()
{
	int n;
	scanf("%d",&n);
	for(int i = 1;i <= n; i++)
		scanf("%d",&num[i]);
	msort(1,n);
	printf("%lld",ans);
	return 0;
} 

树状数组加离散化

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN = 500010;
int a[MAXN],b[MAXN],c[MAXN],ranks[MAXN],n;
struct info{
	int val,num;
	bool operator < (const info & a) const { return val == a.val ? num > a.num : val > a.val;}
}node[MAXN];
int lowbit(int x)  //求2^k k为x末尾1的位置 从零位开始
{
	return x&(-x);
}
void update(int x) //修改原数组的值并修改树状数组的值
{
	b[x]++;
	while(x <= n){
		c[x]++;
		x += lowbit(x);
	}
}
int getsum(int x) //求前缀和 c数组代表树状数组
{
	int s = 0;
	while(x > 0){
		s += c[x];
		x -= lowbit(x);
	}
	return s;
}
void change()  //离散化
{
	sort(node+1,node+n+1);
	for(int i = 1;i <= n; i++){
		ranks[node[i].num] = i; 
	}
}
int main()
{
	scanf("%d",&n);
	for(int i = 1;i <= n; i++){ 
		scanf("%d",&a[i]);  //a[]可以不用 就直接写node[i].val
		node[i].val = a[i];
		node[i].num = i;
	}
	ll ans = 0;
	change();
	for(int i = 1;i <= n; i++){
		update(ranks[i]);
		ans += getsum(ranks[i]-1);
	}
	printf("%lld",ans);
	return 0;
}