[BZOJ3192][JLOI2013]删除物品

最新推荐文章于 2018-09-22 10:53:00 发布

原创最新推荐文章于 2018-09-22 10:53:00 发布 · 572 阅读

0 ·

CC 4.0 BY-SA版权

BZOJ 同时被 2 个专栏收录

159 篇文章

订阅专栏

数据结构_线段树

13 篇文章

订阅专栏

本文介绍了一种使用ZKW线段树解决特定问题的方法，通过计算物品下方优先级分布的连续段数来确定物品需要被移动的次数，并提供了一份详细的AC代码实现。

原题地址

自己想的奇葩做法…
对于一个优先级为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;
}