洛谷 P1908 逆序对

题目描述

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

输入输出格式

输入格式：

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

第二行n个数，表示给定的序列。

输出格式：

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

输入输出样例

输入样例#1：

6
5 4 2 6 3 1

输出样例#1：

11

说明

对于50%的数据，n≤2500

对于100%的数据，n≤40000。

树状数组求逆序对模板题，记得离散化。

#include<algorithm>
#include<iostream>
#include<cstdio>
using namespace std;
const int N=40005;
int n,tp,ans,a[N],c[N],d[N];
int lowbit(int x)
{
	return x&(-x);
}
int query(int pos)
{
	int sum=0;
	while(pos>0)
	{
		sum+=c[pos];
		pos-=lowbit(pos);
	}
	return sum;
}
void add(int pos)
{
	while(pos<=n)
	{
		c[pos]++;
		pos+=lowbit(pos);
	}
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		d[++tp]=a[i];
	}
	sort(d+1,d+tp+1);
	tp=unique(d+1,d+tp+1)-d-1;
	for(int i=1;i<=n;i++)
	{
		ans+=query(tp)-query(lower_bound(d+1,d+tp+1,a[i])-d);
		add(lower_bound(d+1,d+tp+1,a[i])-d);
	}
	printf("%d\n",ans);
	return 0;
}