[BZOJ3192][JLOI2013]删除物品

来源：互联网发布：网络教育有毕业证吗编辑：程序博客网时间：2024/05/16 05:27

原题地址

自己想的奇葩做法…
对于一个优先级为x的物品，考察该物品下方优先级大于x物品的优先级被分成了多少个连续段(举个例子,如x=4,该物品下方物品的优先级有5,6,7,9,10,13,14,19,则段数为4)，则该物品需要被移动的次数k=段数*2-(优先级为x+1的物品在该物品下方?1:0).
用的是ZKW线段树维护信息，略复杂，就当是练手ZKW线段树了…

AC code:

#include <cmath>#include <cstdio>#include <vector>#include <algorithm>using namespace std;const int N=100010;int n1,n2,s;int a1[N],a2[N];long long ans;struct D{    int val,typ,num;    D() {}    D(int val,int typ,int num):val(val),typ(typ),num(num) {}    friend bool operator<(D x,D y){        return x.val<y.val;    }};vector<D> d;void discre(){    for(int i=1;i<=n1;i++) d.push_back(D(a1[i],1,i));    for(int i=1;i<=n2;i++) d.push_back(D(a2[i],2,i));    sort(d.begin(),d.end());    for(int i=0,cnt=1;i<(int)d.size();i++,cnt++){        if(d[i].typ==1) a1[d[i].num]=cnt;        else a2[d[i].num]=cnt;    }}struct nod{    int l,r,tot;};struct Segtree{    nod T[N<<3];    Segtree(){        s=1<<(((int)log2(n1+n2+3))+1);    }    int query(int L,int R){        int sum=0,lp=s+L-1,rp=s+R+1,ll=0,rr=0;        for(;lp+1!=rp;lp>>=1,rp>>=1){            if(!(lp&1)&&T[lp^1].tot){                sum+=T[lp^1].tot;                if(ll&&T[lp^1].l==ll+1) sum--;                if(T[lp^1].r==rr-1) sum--;                ll=T[lp^1].r;            }            if((rp&1)&&T[rp^1].tot){                sum+=T[rp^1].tot;                if(ll&&T[rp^1].l==ll+1) sum--;                if(T[rp^1].r==rr-1) sum--;                rr=T[rp^1].l;            }        }        return sum;    }    void insert(int p){        T[s+p].tot=1;T[s+p].l=T[s+p].r=p;        for(int i=(s+p)>>1;i;i>>=1){            T[i].l=T[i<<1].l?T[i<<1].l:T[(i<<1)^1].l;            T[i].r=T[(i<<1)^1].r?T[(i<<1)^1].r:T[i<<1].r;            T[i].tot=T[i<<1].tot+T[(i<<1)^1].tot;            if(T[i<<1].r&&T[i<<1].r+1==T[(i<<1)^1].l) T[i].tot--;        }    }    void clear(){        for(int i=1;i<(s<<2);i++) T[i].l=T[i].r=T[i].tot=0;    }};int main(){    scanf("%d%d",&n1,&n2);    Segtree T;    for(int i=1;i<=n1;i++) scanf("%d",&a1[i]);    for(int i=1;i<=n2;i++) scanf("%d",&a2[i]);    discre();    for(int i=n1;i>=1;i--){        ans+=T.query(a1[i]+1,n1+n2)<<1;        if(T.T[s+a1[i]+1].tot) ans--;        T.insert(a1[i]);      }    T.clear();    for(int i=n2;i>=1;i--){        ans+=T.query(a2[i]+1,n1+n2)<<1;        if(T.T[s+a2[i]+1].tot) ans--;        T.insert(a2[i]);    }    printf("%lld\n",ans);    return 0;}

0 0