POJ - 2299 Ultra-QuickSort

最新推荐文章于 2024-11-15 23:35:43 发布

原创最新推荐文章于 2024-11-15 23:35:43 发布 · 495 阅读

0 ·

CC 4.0 BY-SA版权

POJ 同时被 2 个专栏收录

20 篇文章

订阅专栏

----树状数组

10 篇文章

订阅专栏

本文介绍了一种利用树状数组求解逆序数的方法，并通过离散化处理大数据范围的问题。通过具体代码实现展示了如何对输入数据进行排序和离散化处理，进而更新树状数组并计算逆序数。

利用树状数组求逆序数，但是数据范围比较大，要使用离散化坐标

参考：http://blog.youkuaiyun.com/guhaiteng/article/details/52138756

// 树状数组求逆序数 + 离散化

#include <cstdio>
#include <string>
#include <algorithm>

#define MAX 500000 + 10

using namespace std;

typedef long long LL;
typedef struct {
    LL val;
    LL pos;
} Node;

Node node[MAX];
LL c[MAX];
LL bit[MAX];

LL lowbit( LL i ) {
    return i & ( -i );
}

void update( LL i, LL x, LL n ) {
    while( i <= n ) {
        bit[i] = bit[i] + x;
        i = i + lowbit( i );
    }
}

LL getSum( LL i ) {
    LL ans = 0;
    while( i > 0 ) {
        ans = ans + bit[i];
        i = i - lowbit( i );
    }

    return ans;
}

bool cmp( Node a, Node b ) {
    return a.val < b.val ? 1 : 0;
}

int main() {
    LL n;
    while( scanf( "%lld", &n ) != EOF ) {
        if( n == 0 ) break;

        for( LL i = 1; i <= n; i++ ) {
            scanf( "%lld", &node[i].val );
            node[i].pos = i;
        }

        sort( node + 1, node + 1 + n, cmp );

        for( LL i = 1; i <= n; i++ ) {  // 离散化
            c[node[i].pos] = i;
        }

        // 初始化树状数组
        for( LL i = 1; i <= n; i++ ) bit[i] = 0;

        LL ans = 0;
        for( int i = 1; i <= n; i++ ) {
            update( c[i], 1, n );
            ans = ans + i - getSum( c[i] );
        }
        printf( "%lld\n", ans );
    }

    return 0;
}