1. 程式人生 > >BZOJ 4939: [Ynoi2016]掉進兔子洞(莫隊+bitset)

BZOJ 4939: [Ynoi2016]掉進兔子洞(莫隊+bitset)

== for memcpy friend 計算 inline 代碼 emc esp

傳送門

解題思路

  剛開始想到了莫隊+\(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;
}

BZOJ 4939: [Ynoi2016]掉進兔子洞(莫隊+bitset)