树状数组求逆序对个数

先给出树状数组的模板：

int lowbit(int a) {
    return a & (-a);
}
void update(int x, int y, int n) {
    for(int i = x; i <= n; i ++)
        c[i] += y;
}
int getSum(int x) {
    int ans = 0;
    for(int i = x; i > 0; i -= lowbit(i))
        ans += c[i];
    return ans;
}

何为逆序对：在一个数组a中，且，则称为一组逆序对。正常情况下，用遍历寻找出一个大小为n的数组中所有的逆序对时间复杂度为。我们这里使用树状数组来解决这个问题。

基本思路是，开一个大小为n+1的数组c，依次遍历原数组，对于每一个a[i]，让c[a[i]]加1，想知道当前状态下有多少逆序对时，只需要算出当前状态下c[a+1, n]中有多少个1，因为这些1代表着出现在a[i]之前却比a[i]大的数字。

不太好的事情是，对于这样一个数列：{1,3,7,6,999}，我们需要开一个大小为1000的数组存放，造成大量空间浪费

这里涉及一个概念——离散化，简单说来就是另开一个数组,d， d[i]用来存放第i大的数在原序列的什么位置，比如原序列a={5，3，4，2，1}，第一大就是5，他在a中的位是1，所以d[1]=1，同理d[2]=3，依次类推，得出d数组为{1，3，2，4，5}。转换之后，空间复杂度大大缩小，我们的问题也转换成了求数组d中的正序对（当数组a全是正序的时候，数组d是逆序的）。

于是我们先进行离散化操作，代码如下：

#include <bits/stdc++.h>
using namespace std;
const int N = 100000;
int a[N], d[N], t[N];
int n;
bool cmp(int x, int y) {
    if(a[x] == a[y]) return x > y; //值相等则按序号大小排
    return a[x] > a[y]; //从大到小排序
}
int main() {
    cin >> n;
    for(int i = 1; i <= n; i ++) {
        cin >> a[i];
        d[i] = i; //初始化d[i]
    }
    sort(d + 1, d + n + 1, cmp);

    return 0;
}

离散化之后，接下来进行求和操作。

由上面的分析，我们在求数组d的正序和，即每次遍历到一个数字x的时候，都要求比他小的元素有几个，而比它小的元素个数一定是1到x中存在数的个数，也就是[1, x-1]中有几个数，有点像之前讲的前缀和，只不过树状数组t表示的是[1,x]中有几个数已经存在，这样我们每次把一个新的数x放进去的时候，都需要把包含这个数的结点更新，然后查询[1,x-1]有几个数已经存在。

#include <bits/stdc++.h>
using namespace std;
const int N = 100000;
int a[N], d[N], t[N];
int n;
int lowbit(int a) {
    return a & -a;
}
int update(int x) { //更新结果
    for(int i = x; i <= n; i += lowbit(i))
        t[i] ++;
}
int getSum(int x) { //查询1~x有几个数加进去了
    int ans = 0;
    for(int i = x; i; i -= lowbit(i))
        ans += t[i];
    return ans;
}
bool cmp(int x, int y) {
    if(a[x] == a[y]) return x > y; //值相等则按序号大小排
    return a[x] > a[y]; //从大到小排序
}
int main() {
    int ans = 0;
    cin >> n;
    for(int i = 1; i <= n; i ++) {
        cin >> a[i];
        d[i] = i; //初始化d[i]
    }
    sort(d + 1, d + n + 1, cmp);
    for(int i = 1; i <= n; i ++) {
        update(d[i]); //放入该数
        ans += getSum(d[i] - 1); //累加
        cout << ans << endl;
    }
    cout << ans;
    return 0;
}