【算法】树状数组 P1908 逆序对

P1908 逆序对

#include<iostream>
#include<cstdio>
#include<algorithm>
#define ll long long
using namespace std;
const int Maxn = 5e5 + 10;
int n;
ll ans = 0;

//树状数组与离散化与逆序对
int tree[Maxn], lisan[Maxn];
struct node {
    int val, num;
}pot[Maxn];//离散化

bool cmp(node a, node b) {
    if (a.val == b.val) {
        return a.num < b.num;
    }
    return a.val < b.val;
}
inline int lowbit(int x) {
    return x & -x;
}
inline void add(int x, int k) {
    while (x <= n) {  // 不能越界
        tree[x] += k;
        x += lowbit(x);
    }
}
inline ll getsum(int x) {
    ll sum = 0;
    while (x >= 1) {
        sum = sum + tree[x];
        x -= lowbit(x);
    }
    return sum;
}

int main()
{
    scanf("%d", &n);
    for (int i = 1; i <= n; i++) {
        scanf("%d", &pot[i].val);//输入世界大小
        pot[i].num = i;
    }
    sort(pot + 1, pot + 1 + n, cmp);
    for (int i = 1; i <= n; i++) {
        lisan[pot[i].num] = i;//离散化,在不改变原数据大小关系和顺序的前提下,变为1到n;
    }

    for (int i = 1; i <= n; i++)
    {
        //注意add(m,n)标示的是在第m项上加n,而不是在前m项上加n
        add(lisan[i], 1);//主要思想是lisan[i]表示第i个数的大小关系,也是第i个数从小到大排第几,将其加一表示这个数已经插入了
        ans += i - getsum(lisan[i]);//查询lisan[i]之前有多少个数已经被插入,因为在树状数组里lisan[i]前面的都是比他小的,并且一定都存在,如果没有插入就表示他在lisan[i]之后插入
        //也就是在原数组中的顺序比lisan[i]靠后,比lisan[i]小,又比他位置靠后,那便是逆序数对了,有多少个呢 =i - getsum(lisan[i])
    }
    //整体思想离散化后1到n都一定有数,而根据原数组排序依次插入树状数组中,lisan[i]=m;在树状数组在m前面的数已经插入的表示原数组在m前面,还没插入的表示在m后面,而在m后面的便一定是
    //逆序对
    printf("%lld\n", ans);
    return 0;
}