【BZOJ3333】【排队计划】（树状数组+线段树）

Description

Input

Output

Sample Input

6 2

160 163 164 161 167 160

2

3

Sample Output

6

3

1

HINT

题解：
题目大意：给定一个序列，每次选择一个位置，把这个位置之后所有小于等于这个数的数抽出来，排序，再插回去，求每次操作后的逆序对数。
很容易看出，只有排序的这些数字的相对位置发生了变化，因此也只有这些数字之间的逆序对个数发生了变化。每次排序，减少的都是选出来的那些数往后的逆序对数。所以我们对所有数建一棵线段树，每次查一下选出数往后的最小值，减去对应的逆序对数，把这个数赋成正无穷，直到查出来的数等于选的那个数。
感谢PoPoQQQ大神的博客。

代码如下：

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<math.h>
#define ll long long
#define inf 0x7f7f7f7f
#define N 500005
#define lb(x) (x&-x)
#define ls (rt<<1)
#define rs (rt<<1|1)
#define mid (l+r>>1)
using namespace std;
ll read()
{
    ll x=0,f=1;
    char c=getchar();
    while(c<'0' || c>'9') {if(c=='-') f=-1;c=getchar();}
    while(c<='9' && c>='0') {x=x*10+c-'0';c=getchar();}
    return x*f;
}
int n,m,a[N],b[N],p[N],cnt,x;
ll f[N],t[N<<2],ans;
int tree[N<<2];
bool cmp(int x,int y) {return b[x]<b[y];}
void add(int x,ll v)
{
    while(x<=cnt)
    {
        t[x]+=v;
        x+=lb(x);
    }
}
ll query(int x)
{
    ll ret=0;
    while(x)
    {
        ret+=t[x];
        x-=lb(x);
    }
    return ret;
}
void pushup(int rt)
{
    if(a[tree[rs]]<=a[tree[ls]]) tree[rt]=tree[rs];
    else tree[rt]=tree[ls];
}
void buildt(int rt,int l,int r)
{
    if(l==r)
    {
        tree[rt]=l;
        return ;
    }
    buildt(ls,l,mid);buildt(rs,mid+1,r);
    pushup(rt);
}
int pd(int x,int y)
{
    if(a[y]<=a[x]) return y;
    return x;
}
int query(int rt,int l,int r,int L,int R)
{
    if(L<=l && r<=R) return tree[rt];
    int ans=0;
    if(L<=mid) ans=query(ls,l,mid,L,R);
    if(R>mid)
    {
        int t=query(rs,mid+1,r,L,R);
        if(!ans) ans=t;
        else ans=pd(ans,t);
    }
    return ans;
}
void mdy(int rt,int l,int r,int t)
{
    if(l==r)
    {
        a[l]=inf;
        f[l]=0;
        return ;
    }
    if(t<=mid) mdy(ls,l,mid,t);
    else mdy(rs,mid+1,r,t);
    pushup(rt);
}
int main()
{
    n=read(),m=read();
    for(int i=1;i<=n;i++) b[i]=read(),p[i]=i;
    sort(p+1,p+n+1,cmp);
    for(int i=1;i<=n;i++)
    {
        if(b[p[i]]!=b[p[i-1]]) a[p[i]]=++cnt;
        else a[p[i]]=cnt;
    }
    for(int i=n;i>=1;i--)
    {
        add(a[i],1);
        f[i]=query(a[i]-1);
        ans+=f[i];
    }
    buildt(1,1,n);
    printf("%lld\n",ans);
    for(int i=1;i<=m;i++)
    {
        x=read();
        while(a[x]!=inf)
        {
            int t=query(1,1,n,x,n);
            ans-=f[t];
            mdy(1,1,n,t); 
        }
        printf("%lld\n",ans);
    }
    return 0;
}