bzoj3295: [Cqoi2011]动态逆序对

Description

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

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)。

Solution

这是一道类似三维偏序的题目
我们先给每个数设定一个t值，即最先被删除的t值最大，没有被删除的点的t值按照位置从小到大拍
那么我们现在得到了一个三元组(t,x,pos)
我们要求的就是t

Code

#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fd(i,a,b) for(i=a;i>=b;i--)
using namespace std;
const int N=100005;
typedef long long ll;
struct edge{
 ll x,y,z;  
} a[N];
ll n,m,i,cnt,x,ans[N],tr[N];
ll read(){
    ll sum=0;
    char c=getchar();
    while (c<'0'||c>'9') c=getchar();
    while (c>='0'&&c<='9'){
        sum=sum*10+c-'0';
        c=getchar();}
    return sum;
} 
bool cmp1(edge a,edge b){
    return a.z<b.z;
}
bool cmp2(edge a,edge b){
    return a.x>b.x;
}
void add(ll x,ll y){
    while (x<=n){
        tr[x]+=y;
        x+=x&(-x);}
}
ll query(ll x){
    ll s=0;
    while (x>0) {
        s+=tr[x];
        x-=x&(-x);}
    return s;
}
void cdq(ll l,ll r){
    if (l==r) return;
    ll mid=(l+r)>>1;
    cdq(l,mid),cdq(mid+1,r);
    ll pl,pr;
    pl=l,pr=mid+1;
    sort(a+l,a+pr,cmp2);
    sort(a+pr,a+r+1,cmp2);
    while (pr<=r){
        while (pl<=mid&&a[pr].x<a[pl].x) 
            add(a[pl].y,1),pl++;
        ans[a[pr].z]+=query(a[pr].y);
        pr++;
    }
  //  if (pl>mid) pl--;
    while (--pl>=l) add(a[pl].y,-1);
    pl=mid,pr=r;
    while (pr>=mid+1){
        while (pl>=l&&a[pr].x>a[pl].x) 
            add(a[pl].y,1),pl--;
        ans[a[pr].z]+=query(n)-query(a[pr].y);
        pr--;
    }
//  if (pl<l) pl++;
    while (++pl<=mid) add(a[pl].y,-1);
}
int main(){
    //freopen("1.in","r",stdin);
    //freopen("1.out","w",stdout);
    n=read(),m=read(),cnt=1;
    fo(i,1,n) x=read(),a[x].x=x,a[x].y=i;
    fo(i,1,m) {
        x=read();
        a[x].z=n-i+1;}
    fo(i,1,n)
      if (!a[i].z) a[i].z=cnt++;
    sort(a+1,a+n+1,cmp1);
    cdq(1,n);
    fo(i,2,n) ans[i]=ans[i]+ans[i-1];
    fd(i,n,n-m+1) printf("%lld\n",ans[i]); 
}