基本思想和线段树求解逆序数是一样的,前一篇《求逆序对 线段树版》也介绍过,先对输入数组离散,数组里的元素都不相同可以直接hash,存在相同的数话可以采用二分。
离散化后对于每个f[i],找到f[i]+1~ n中的个数,也就是到i这个位置,一共有多少比f[i]大的数,统计之后在将f[i]的位置上的数量加1。
这样一来统计的就是类似a[i]~n的和,可以想象成 把树状数组反过来统计,即统计的时候加lowbit,更新的时候减lowbit。
还是 以POJ 2299为例。
#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <set>
#include <stack>
#include <cctype>
#include <algorithm>
#define lson o<<1, l, m
#define rson o<<1|1, m+1, r
using namespace std;
typedef long long LL;
const int maxn = 500050;
const int mod = 99999997;
const int MAX = 0x3f3f3f3f;
int n, tmp, f[maxn];
LL c[maxn];
struct C {
int num, pos;
}in[maxn];
bool cmp(C x, C y) {
return x.num < y.num;
}
int main()
{
while(cin >> n, n) {
for(int i = 1; i <= n; i++) {
scanf("%d", &in[i].num);
in[i].pos = i;
}
sort(in+1, in+1+n, cmp);
LL sum = 0;
memset(c, 0, sizeof(c));
for(int i = 1; i <= n; i++) f[ in[i].pos ] = i;
for(int i = 1; i <= n; i++) {
for(int j = f[i]+1; j <= n; j += j&(-j))
sum += c[j];
for(int j = f[i]; j > 0; j -= j&(-j))
c[j]++;
}
cout << sum << endl;
}
return 0;
}