poj 2299 Ultra-QuickSort(fenwick树求逆序对)

最新推荐文章于 2022-12-15 12:59:42 发布

slowlight93

最新推荐文章于 2022-12-15 12:59:42 发布

阅读量562

点赞数

分类专栏： ACM.数据结构文章标签：树状数组 fenwick树逆序对

本文链接：https://blog.csdn.net/slowlight93/article/details/39931061

版权

ACM.数据结构专栏收录该内容

47 篇文章 0 订阅

订阅专栏

树状数组求逆序对数方法 GET！

题意很直白：求一个长度为n的序列a1,a2,a3...an的逆序对数，其中aj在区间[0, 999 999 999]内。

假设一个数组x, x[aj]表示aj存不存在。

数组c是x的fenwick树, 将序列元素依次插入c。

如果当前插入aj，sum(aj)表示已经插入的前j个数中，小于等于aj的数的个数

，逆序对增加j-sum(aj)

所以先清空c，然后依次插入，累加答案。

但注意到aj的区间很大，所以先离散化一下。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <vector>
#include <queue>
#include <stack>
#include <cassert>
#include <algorithm>
#include <cmath>
#include <set>
#include <list>
#include <map>
#include <limits>
using namespace std;

#define MIN(a, b) ((a) < (b) ? (a) : (b))
#define MAX(a, b) ((a) > (b) ? (a) : (b))
#define REP(i, s, t) for(int (i)=(s);(i)<=(t);++(i))
#define UREP(i, s, t) for(int (i)=(s);(i)>=(t);--(i))
#define REPOK(i, s, t, o) for(int (i)=(s);(i)<=(t) && (o);++(i))

#define MAXN 500000
#define MAXM 10000
#define MOD 10000007

#define PI 3.1415926535897932384626433832795
#define HALF_PI 1.5707963267948966192313216916398

typedef long long LL;
typedef vector<int> veci;
typedef vector<pair<int, int> > vect;
typedef pair<int, int> pairi;

const double maxdouble = numeric_limits<double>::max();
const double eps = 1e-10;
const int INF = 0x7FFFFFFF;

struct node{
    int val;
    int pos;
};

node arr[MAXN+5];
int re[MAXN+5];
int c[MAXN + 5];
int n;
/*
struct comp_obj{
    bool operator() (const node & lhs, const node & rhs) {
        return lhs.val < rhs.val;
    }
};*/
bool comp(const node & lhs, const node & rhs) {
    return lhs.val < rhs.val;
}

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

int sum(int x) {
    int ret = 0;
    while(x > 0) {
        ret += c[x];
        x -= lowbit(x);
    }
    return ret;
}

void update(int x) {
    while(x <= n) {
        c[x] += 1;
        x += lowbit(x);
    }
}

int main() {
    //freopen("input.in", "r", stdin);

    while(scanf("%d",&n) == 1 && n) {
        REP(i, 1, n) {
            scanf("%d", &arr[i].val);
            arr[i].pos = i;
        }
        sort(arr+1, arr+n+1, comp);
        REP(i, 1, n) {
            re[arr[i].pos] = i;
        }
        REP(i, 0, n)
            c[i] = 0;
        LL ans = 0;
        REP(i, 1, n) {
            update(re[i]);
            ans += i - sum(re[i]);
        }
        cout << ans << endl;
    }

    return 0;
}