P1908 逆序对

题目链接

luogu P1908

题面

题目描述

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

Update:数据已加强。

输入格式

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

输出格式

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

样例 #1

样例输入 #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$

题意

有 $n$ 个元素存入 $num$ 数组里，依次以 $1\sim n$ 编号
当 $i<j$ 且 $num[i]>num[j]$ 成立，责任为 $(i,j)$ 为一对逆序对
求 $num$ 数组里一共有多少对逆序对

思路

首先，我们要倒序处理 $num$ 数组
并且，对于每一个 $num[i]$ ，在 $num[i+1\sim n]$内比它小的元素都可以组成逆序对
所以我们还要每次求$num[i+1\sim n]<num[i]$ 的个数
如果用双重循环枚举，那肯定要 $TLE$

---

我们转换一下思路，比 $num[i]$ 小的数有 $num[i]-1$ 个
只要我们建立一个标记数组，记录一下遇到的所有元素的数量
每次求 $num[i]$ 的逆序对时就对标记数组 $1\sim num[i]-1$ 进行累加，求出的值便是 $num[i]$ 所在的逆序对个数

---

可是上述方法也是 $O(n^2)$ 的时间复杂度（查询用了一重循环），并且标记数组也开不了这么大
那么该怎么处理呢？

离散化——压缩$num$数组

因为每个元素的上限是$10^9$，若对每一个整数都给予一个标记数组的空间
那会有很多空余，很浪费
这时我们可以用离散化压缩$num$数组的元素，使标记数组的大小缩减到$10^6$

树状数组——$O(lo{g}_2^n)$查询+标记

在离散化之后，倒序遍历$num$数组
先用树状数组的查询 $(query)$ 函数，查询一下值为 $1\sim num[i]-1$ 的元素之前出现的次数，即为 $num[i]$ 所在的逆序对对数
然后再给 $num[i]$ 在树状数组里面的对应区间更新一下标记即可
细节见代码

#include<bits/stdc++.h>
using namespace std;
#define int long long
int n,rak[500010],num[500010],ans,c[500010];
int lowbit(int x){
	return x&-x;
}
void add(int x,int k){//树状数组单点修改 
	while(x<=n){
		c[x]+=k;
		x+=lowbit(x);
	}
	return;
}
int query(int x){//树状数组查询前缀和 
	int sum=0;
	while(x>0){
		sum+=c[x];
		x-=lowbit(x);
	}
	return sum;
}
signed main(){
	scanf("%lld",&n);
	for(int i=1;i<=n;i++)scanf("%lld",&num[i]),rak[i]=num[i];
	sort(rak+1,rak+1+n);//离散化排序 
	int new_n=unique(rak+1,rak+1+n)-(rak+1);//离散化去重 
	for(int i=1;i<=n;i++)num[i]=lower_bound(rak+1,rak+1+n,num[i])-rak;//离散化赋值新下标 
	for(int i=n;i>=1;i--){
		ans+=query(num[i]-1);//统计i+1~n中有几个元素比num[i]大，所得值即为num[i]所在的逆序对个数 
		add(num[i],1);//标记++ 
	}
	printf("%lld",ans);
	return 0;
}