题目链接;http://poj.org/problem?id=2299
题目大意: 给出n个数 求他逆序对数的对数
思路:
1.离散化+树状数组
2 题目的输入个数有500000的规模但是每个数的最大值为999999999,因此我们需要离散化这些数据
3 对于数据9 1 0 5 4我们离散化成5 2 1 4 3
那么对于输入一个树a[i]我们去求一下它的离散化后的id,然后去求前面比这个id大的个数
4 由于getSum(x)函数的求和是求[1,x]而不是[x,MAXN),所以我们可以换成求小于等于id的个数即getSum(id),
然后i-1-getSum(id)就是比id大的个数,最后在更新一下treeNum[id]
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int MAXN = 500010;
int n;
int tmpNum[MAXN] , num[MAXN];
int treeNum[MAXN];
int lowbit(int x){
return x&(-x);
}
int getSum(int x){
int sum = 0;
while(x){//将以x为根节点的数 相当于求其叶节点的和 但其非叶节点等于其两个子节点的和
sum += treeNum[x];
x -= lowbit(x);
}
return sum;
}
void add(int x , int val){
while(x < MAXN){//相当于一个树 他将其自己以及他的父节点一路向上都加val
treeNum[x] += val;
x += lowbit(x);
}
}
long long solve(){
long long ans = 0;
memcpy(tmpNum , num , sizeof(num));//复制
memset(treeNum , 0 , sizeof(treeNum));
sort(num+1 , num+1+n); //排序 去重 离散化
int len = unique(num+1 , num+1+n)-(num+1);
for(int i = 1 ; i <= n ; i++){
int id = lower_bound(num+1,num+1+len,tmpNum[i])-num;
ans += i-getSum(id)-1;//getsum是指在此之前比他小或者等于的数的个数 i-getsum-1是比他大的个数
add(id , 1);//相当于一个树 他将其自己以及他的父节点一路向上都加1
}
return ans;
}
int main(){
while(scanf("%d" , &n) && n){
for(int i = 1 ; i <= n ; i++)
scanf("%d" , &num[i]);
printf("%lld\n" , solve());
}
return 0;
}