BZOJ 4939: [Ynoi2016]掉進兔子洞(莫隊+bitset)
阿新 • • 發佈:2019-03-15
== 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)