树状数组求逆序对

洛谷 P1908（逆序对）

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

Problem Description

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

Input

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

Output

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

Sample Input

6
5 4 2 6 3 1

Sample Output

11

思路

这是一道利用树状数组求逆序对的对数的板子题。
首先自定义一个结构体，这个结构体中存放值和这个值的编号。同时自定义排序的规则，即若值相同则编号靠前的排在前，否则值越小的越前。
以输入样例为例：
$node$数组经过排序之后为下面的状态
$val⟹123456$
$idx⟹635214$
然后从第$1$个遍历到第$6$个
以遍历到第$3$个元素为例
此时我们可以认为有一个数组 a [ 7 ] = { 0 , 0 , 0 , 1 , 0 , 0 , 1 } a[7] = \{0, 0, 0, 1, 0, 0, 1\} a[7]={0,0,0,1,0,0,1}。（有$7$个数是因为数组下标从$0$开始的）
add(nn[i].idx, 1, c)（此时$nn[i].idx$为$5$）可以看作在$a[5]$上加$1$，$a$数组则变为 { 0 ,   0 ,   0 ,   1 ,   0 ,   1 ,   1 } \{0,\ 0,\ 0,\ 1,\ 0,\ 1,\ 1\} {0, 0, 0, 1, 0, 1, 1}。
sum(nn[i].idx, c)用来得到$a$数组前$nn[i].idx$项的和，此时$nn[i].idx$为$5$，得到的值就是$2$，这也就是说在第$5$个位置上的数前面有$2$个数是小于等于它的。（包括其自身）
nn[i].idx-sum(nn[i].idx, c)第$3$个数位于第$5$个位置，而第$5$个位置上的数前面有$2$个数是小于等于它的，那么这个数对答案的贡献也就是$5-2=3$啦。

代码

#include <bits/stdc++.h>
#define pi acos(-1.0)
#define ll long long
#define ull unsigned long long
#define esp 1e-9
#define inf 0x3f3f3f3f
#define inff 0x3f3f3f3f3f3f3f3f
#define Pair pair<ll, ll>
#define It list<node>::iterator
   
using namespace std;

typedef struct{
	ll val, idx;
} node;

bool cmp(node a, node b){
	if (a.val == b.val){
		return a.idx<b.idx;
	}
	return a.val<b.val;
}

const ll N = 5e5+5;
ll n, ans = 0, c[N];
node nn[N];

ll lowbit(ll x){
	return x&(-x);
}

void add(ll pos, ll val, ll arr[]){//对arr[pos]进行加val的单点更新
	while (pos < N){
		arr[pos] += val;
		pos += lowbit(pos);
	}
}

ll sum(ll pos, ll arr[]){//求arr数组中下标为1~pos所有元素的和
	ll res = 0;
	while (pos){
		res += arr[pos];
		pos -= lowbit(pos);
	}
	return res;
}

int main(){
	ios::sync_with_stdio(false);
	cin>>n;
	for (ll i = 1; i <= n; i++){
		cin>>nn[i].val;
		nn[i].idx = i;
	}
	sort(nn+1, nn+n+1, cmp);
	memset(c, 0, sizeof(c));
	for (ll i = 1; i <= n; i++){
		add(nn[i].idx, 1, c);
		ans += (nn[i].idx-sum(nn[i].idx, c));
	}
	cout<<ans<<endl;
	
	return 0;
}