动态逆序对

题目链接

Description

对于序列A，它的逆序对数定义为满足i<j，且Ai>Aj的数对(i,j)的个数。给1到n的一个排列，按照某种顺序依次删
除m个元素，你的任务是在每次删除一个元素之前统计整个序列的逆序对数

Input

输入第一行包含两个整数n和m，即初始元素的个数和删除的元素个数。
以下n行每行包含一个1到n之间的正整数，即初始排列。
以下m行每行一个正整数，依次为每次删除的元素。
N<=100000 M<=50000

Output

输出包含m行，依次为删除每个元素之前，逆序对的个数。

Sample Input

5 4
1
5
3
4
2
5
1
4
2

Sample Output

5
2
2
1
样例解释
(1,5,3,4,2)(1,3,4,2)(3,4,2)(3,2)(3)。
Hint
Source

思考

在网上看了许多的题解发现有多种做法， 但主要分为两类一种为CDQ分治， 一种为树套树的做法。其中树套树中，树状数组套线段树的代码是最为简洁易懂的但缺点也很明显，就是所需要的空间较大。其具体做法为在树状数组中的每个点上都开一个线段树（维护树状数组上对应的区间的数， 线段树上的每个点维护对应值区间内数的多少），这样便于查询在树状数组上的区间l ~ r中比x小的数字有多少，又由于所需空间较大故采取动态开点，细节见于代码。

#include <bits/stdc++.h>

using namespace std;
#define lowbit(x) x & -x
#define ll long long
const int N = 1e5 + 10;
const int M = 9e6 + 10;
int n, m;
int ls[M], rs[M], tr[M], pos[N], cnt; // tr[i]第i个区间中的数字的数量， pos[i], 值 i 为的数在树状数组中的位置
int T[N]; // 树状数组
ll ans;

void insert_(int &k, int l, int r, int x) 
{
    if(k == 0) k = ++cnt;
    tr[k] ++;
    if(l == r) return;
    int mid = l + r >> 1;
    if(x <= mid) insert_(ls[k], l, mid, x);
    else insert_(rs[k], mid + 1, r, x);
}
int query(int k, int l, int r, int x, int y)
{
    if(x <= l && r <= y) return tr[k];
    int mid = l + r >> 1;
    int sum = 0;
    if(x <= mid) sum += query(ls[k], l, mid, x, y);
    if(mid < y) sum += query(rs[k], mid + 1, r, x, y);
    return sum;
}

void update(int k, int l, int r, int x)
{
    if(k == 0) return; // 如果这个区间还没有开点数码没有数
    tr[k] --;
    if(l == r) return;
    int mid = l + r >> 1;
    if(mid >= x) update(ls[k], l, mid, x);
    else update(rs[k], mid + 1, r, x);
}
int main()
{
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; i++)
    {
        int x;
        scanf("%d", &x);
        pos[x] = i;
        for(int j = i; j <= n; j += lowbit(j)) insert_(T[j], 1, n, x); // 在第j棵线段树中插入x
        if(x != n)
        for(int j = i; j >= 1; j -= lowbit(j)) ans += (ll)query(T[j], 1, n, x + 1, n);// 查询比1 ~ i 中比x小的数
    }
    while(m --)
    {
        int x;
        scanf("%d", &x);
        printf("%lld\n", ans);
        if(x != n) for(int i = pos[x]; i >= 1; i -= lowbit(i)) ans -= (ll)query(T[i], 1, n, x + 1, n); // 减去在x的前面并比x大的数
        if(x != 1)
        {
            for(int i = n; i >= 1; i -= lowbit(i)) ans -= (ll)query(T[i], 1, n, 1, x - 1); // 减去整个区间中比x小的数
            for(int i = pos[x] - 1; i >= 1; i -= lowbit(i)) ans += (ll)query(T[i], 1, n, 1, x  - 1);// 加上在x前面并比x小的数
            // 这两步实现了ans 减去在x后面并比x小的数
        }
        for(int i = pos[x]; i <= n; i += lowbit(i)) update(T[i], 1, n, x);
    }
    return 0;
}