BZOJ3295 [Cqoi2011]动态逆序对 【CDQ分治】

题目

对于序列A，它的逆序对数定义为满足i

输入格式

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

输出格式

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

输入样例

5 4

1

5

3

4

2

5

1

4

2

输出样例

5

2

2

1

样例解释

(1,5,3,4,2)(1,3,4,2)(3,4,2)(3,2)(3)。

提示

N<=100000 M<=50000

题解

反过来插入元素
对于每个元素，对应一个三元组(t,x,y)，表示t时刻插入，位置x，权值y
每个元素插入时产生的逆序对数量为所有的满足如下条件的(t’,x’,y’)
t’ < t 且 x’ > x 且 y’ <ｙ
或者　t’ < t 且 x’ < x 且 y’ > y
这就是三维偏序，可以用CDQ分治
因为有两种情况，所以要正反统计两次
【压行代码50行不到，CDQ很感人】

#include<iostream>
#include<cstdio>
#include<algorithm>
#define LL long long int
#define REP(i,n) for (int i = 1; i <= (n); i++)
#define lbt(x) (x & -x)
using namespace std;
const int maxn = 100005,maxm = 100005,INF = 1000000000;
inline int read(){
    int out = 0,flag = 1; char c = getchar();
    while (c < 48 || c > 57) {if (c == '-') flag = -1; c = getchar();}
    while (c >= 48 && c <= 57) {out = (out << 3) + (out << 1) + c - '0'; c = getchar();}
    return out * flag;
}
struct Que{int t,x,y;}Q[maxn],T[maxn];
inline bool operator <(const Que& a,const Que& b){return a.x > b.x;}
int N,M,A[maxn],B[maxn],C[maxn],cnt,Qi,s[maxn];
LL ans[maxn];
void add(int u,int v){while (u <= N) s[u] += v,u += lbt(u);}
int query(int u){int Ans = 0; while (u) Ans += s[u],u -= lbt(u); return Ans;}
void cdq(int l,int r){
    if (l == r) return;
    int mid = l + r >> 1,l1 = l,l2 = mid + 1;
    for (int i = l; i <= r; i++)
        if (Q[i].t <= mid) add(Q[i].y,1);
        else ans[Q[i].t] += query(Q[i].y);
    for (int i = l; i <= r; i++)
        if (Q[i].t <= mid) add(Q[i].y,-1);
    for (int i = r; i >= l; i--)
        if (Q[i].t <= mid) add(N - Q[i].y + 1,1);
        else ans[Q[i].t] += query(N - Q[i].y + 1);
    for (int i = l; i <= r; i++)
        if (Q[i].t <= mid) T[l1++] = Q[i],add(N - Q[i].y + 1,-1);
        else T[l2++] = Q[i];
    for (int i = l; i <= r; i++) Q[i] = T[i];
    cdq(l,mid); cdq(mid + 1,r);
}
int main(){
    cnt = N = read(); M = read();
    REP(i,N) B[A[i] = read()] = i;
    REP(i,M){C[Q[i].x = B[Q[i].y = read()]] = true; Q[i].t = cnt--; ++Qi;}
    REP(i,N) if (!C[i]) Q[++Qi] = (Que){cnt--,i,A[i]};
    sort(Q + 1,Q + 1 + N);
    cdq(1,N);
    REP(i,N) ans[i] += ans[i - 1];
    for (int i = 0; i < M; i++) printf("%lld\n",ans[N - i]);
    return 0;
}