BZOJ 4939: [Ynoi2016]掉进兔子洞(莫队+bitset)

传送门

解题思路

  刚开始想到了莫队+\(bitset\)去维护信息,结果发现空间不太够。。试了各种奇技淫巧都\(MLE\),最后\(\%\)了发题解发现似乎可以分段做。。这道题做法具体来说就是开\(3\)个\(bitset\),然后对原序列离散化之后给每个值规定一个开始的位置,之后就可以莫队搞,计算答案是用总的元素个数减去扔掉的,而扔掉的其实就是三个\(bitset\)做与运算后\(1\)的个数,时间复杂度\(O(n\sqrt n+\frac{n^2}{w})\)。\(bzoj\)上时限\(80s\)跑了\(80s\),荣获倒一。

代码

#include<bits/stdc++.h>

using namespace std;
const int N=100005;
const int M=10000;

inline int rd(){
    int x=0,f=1; char ch=getchar();
    while(!isdigit(ch)) f=ch=='-'?0:1,ch=getchar();
    while(isdigit(ch)) x=(x<<1)+(x<<3)+ch-'0',ch=getchar();
    return f?x:-x;
}

int n,m,a[N],cpy[N],pos[N],u,b[N],siz,num,slen[N];
int ll[N][4],rr[N][4];

struct Data{
    int l,r,id,type;
    friend bool operator<(const Data A,const Data B){
        if(A.l/siz!=B.l/siz) return A.l<B.l;
        if((A.l/siz)&1) return A.r>B.r;
        return A.r<B.r;
    }
    Data(int _l=0,int _r=0,int _id=0,int _type=0){
        l=_l; r=_r; id=_id; type=_type;
    }
}q[N*3];
bitset<N> f[3][10010],g;

#define Add(x) g.set(pos[a[(x)]]++)
#define Del(x) g.reset(--pos[a[(x)]])

void solve(int l,int r){
    int tot=0;
    for(int i=1;i<=n;i++)
        if(b[i]!=b[i-1]) pos[b[i]]=i;
    for(int i=l;i<=r;i++){
        q[++tot]=Data(ll[i][0],rr[i][0],i-l+1,0);
        q[++tot]=Data(ll[i][1],rr[i][1],i-l+1,1);
        q[++tot]=Data(ll[i][2],rr[i][2],i-l+1,2);
    }
    sort(q+1,q+1+tot);
    int L=1,R=0; g.reset();
    for(int i=1;i<=tot;i++){
        while(L>q[i].l) L--,Add(L);
        while(R<q[i].r) R++,Add(R);
        while(L<q[i].l) Del(L),L++;
        while(R>q[i].r) Del(R),R--;
        f[q[i].type][q[i].id]=g;
    }   
    for(int i=1;i<=r-l+1;i++){
        printf("%d\n",(slen[i+l-1]-3*((f[0][i]&f[1][i]&f[2][i]).count()))); 
        f[0][i].reset(); f[1][i].reset(); f[2][i].reset();
    }
}

int main(){
    n=rd(),m=rd(); int l1,r1,l2,r2,l3,r3;
    for(int i=1;i<=n;i++) a[i]=cpy[i]=rd();
    sort(cpy+1,cpy+1+n); siz=sqrt(n); 
    u=unique(cpy+1,cpy+1+n)-cpy-1;
    for(int i=1;i<=n;i++)
        a[i]=lower_bound(cpy+1,cpy+1+u,a[i])-cpy;
    memcpy(b,a,sizeof(b)); sort(b+1,b+1+n);
    for(int i=1;i<=m;i++){
        l1=rd(),r1=rd(),l2=rd(),r2=rd(),l3=rd(),r3=rd();
        ll[i][0]=l1,ll[i][1]=l2,ll[i][2]=l3;
        rr[i][0]=r1,rr[i][1]=r2,rr[i][2]=r3;
        slen[i]=r1-l1+1+r2-l2+1+r3-l3+1;
    }
    for(int i=1;i<=m;i+=M)
        solve(i,min(m,i+M-1));
    return 0;
}
上一篇:Java BitSet使用场景和示例


下一篇:Cowpatibility —— bitset找与某个数组有任何一个元素相交的所有数组