POJ 2299 Ultra-QuickSort(树状数组入门) 求逆序数

/**
*   树状数组 入门题: (求逆序数)
*       这道题wa了好几次,还是意识不够!看discuss才知道,答案是要用long long型来存的!
*   还有个就是离散化,相当于把所有数据的值先排个序,换成另外一个1~n的值。
*       怎么说呢? 比如  9 5 1 3 8 离散化后就是 5 3 1 2 4 这样尽管单个数据再大,
*   也能根据大小离散成 1~n的顺序。 目的是为了能使 bit[i] 表示对于数i之前的数已出现的个数。
*   若不进行离散化的话, i 会很大,导致数组也开不下。
*/

#include <cstdio>
#include <cstring>
#include <cmath>
#include <string>
#include <algorithm>
#define INF 0x7fffffff
#define MAXS 500005
#define LL __int64
using namespace std;
struct Node {
    LL d;
    int pos;
    bool operator < (const Node &a) const {
        return d < a.d;
    }
}num[MAXS];
int n, c[MAXS], bit[MAXS];


void input() {
    for(int i = 1; i <= n; i ++) {
        scanf("%I64d", &num[i].d);
        num[i].pos = i;
    }
}

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

void change() {
    for(int i = 1; i <= n; i ++)
        c[num[i].pos] = i;
}

int get_sum(int x) {
    int ret = 0;
    for(int i = x; i > 0; i -= lowbit(i)) {
        ret += bit[i];
    }
    return ret;
}

void update(int d) {
    for(int i = d; i <= n; i += lowbit(i)) {
        bit[i] ++;
    }
}

LL cal_ans() {
    LL ret = 0;
    for(int i = 1; i <= n; i ++) {
        ret += c[i] - get_sum(c[i] - 1) - 1;
        update(c[i]);
    }
    return ret;
}

int main()
{
    while(scanf("%d", &n), n) {
        memset(bit, 0, sizeof(bit));
        input();
        sort(num + 1, num + n + 1);
        change();
        printf("%I64d\n", cal_ans());
    }
    return 0;
}


 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值