1. 程式人生 > >bzoj3920: Yuuna的禮物(莫隊+分塊套分塊)

bzoj3920: Yuuna的禮物(莫隊+分塊套分塊)

max ons del closed 感覺 做到 bzoj ace clu

  思路挺簡單的,但是總感覺好難寫...碼力還是差勁,最後寫出來也挺醜的

  這題顯然是個莫隊題,考慮怎麽轉移和詢問...

  根據莫隊修改多查詢少的特點,一般用修改快查詢慢的分塊來維護。查第$k_1$小的出現次數可以用權值分塊做到$O(1)$修改,$O(\sqrt{n})$查詢,$k_2$小的數同理。對於每一種出現次數$i$,預處理出有幾種數在序列裏的出現次數$\geq i$,並在每種出現次數中對這些數離散化,這樣我們就能對每種出現次數進行權值分塊查第$k_2$小的數了。

  因為$\sum cnt_i=n$,所以空間是$O(n)$的,這題卡空間...$O(n\sqrt{n})$的空間過不了....

技術分享圖片
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<vector>
#include<cmath>
#include<algorithm>
#define ll long long
using namespace std;
const int maxn=40010, inf=1e9, sqrtm=210;
struct poi{int l, r, k1, k2, pos;}q[maxn];
int n, m, x, blo; vector<int>v[maxn], v2[maxn], sum2[maxn], blsum2[maxn]; int sum1[maxn], blsum1[sqrtm], a[maxn], b[maxn], bl[maxn], cnt[maxn], ans[maxn]; inline void read(int &k) { int f=1; k=0; char c=getchar(); while(c<0 || c>9) c==-&&(f=-1), c=getchar();
while(c<=9 && c>=0) k=k*10+c-0, c=getchar(); k*=f; } bool operator < (poi a, poi b) {return bl[a.l]<bl[b.l] || (bl[a.l]==bl[b.l] && ((bl[a.l]&1)?a.r<b.r:a.r>b.r));} inline void add(int x, int delta) { x=a[x]; int pos=v2[x][cnt[x]-1]; sum1[cnt[x]]+=delta; if(sum1[cnt[x]]==1 && delta==1) blsum1[bl[cnt[x]]]++; if(sum1[cnt[x]]==0 && delta==-1) blsum1[bl[cnt[x]]]--; sum2[cnt[x]][pos]+=delta; blsum2[cnt[x]][bl[pos]]+=delta; } inline void update(int x, int delta) { if(cnt[a[x]]>0) add(x, -1); cnt[a[x]]+=delta; if(cnt[a[x]]>0) add(x, 1); } inline int query1(int k) { int x=0, cnt=0; for(int i=1;i<=bl[n];i++) if(cnt+blsum1[i]>=k) break; else cnt+=blsum1[i], x++; for(int i=blo*x;i<=min(n, blo*(x+1));i++) if(cnt+(sum1[i]!=0)>=k) return i; else cnt+=(sum1[i]!=0); return 0; } inline int query2(int ty, int k) { int x=0, cntt=0; for(int i=1;i<blsum2[ty].size();i++) if(cntt+blsum2[ty][i]>=k) break; else cntt+=blsum2[ty][i], x++; for(int i=blo*x;i<=min(n, blo*(x+1));i++) { if(cntt+sum2[ty][i]>=k) return i; else cntt+=sum2[ty][i]; } return 0; } int main() { read(n); blo=sqrt(n); for(int i=0;i<=n;i++) bl[i]=i/blo+1; for(int i=1;i<=n;i++) read(a[i]), b[i]=a[i]; sort(b+1, b+1+n); for(int i=1;i<=n;i++) cnt[b[i]]++; for(int i=1;i<=n;i++) { for(int j=1;j<=cnt[b[i]];j++) { v2[b[i]].push_back(v[j].size()); v[j].push_back(b[i]); sum2[j].push_back(0); } cnt[b[i]]=0; } for(int i=1;i<=n;i++) for(int j=0;j<=bl[sum2[i].size()];j++) blsum2[i].push_back(0); read(m); for(int i=1;i<=m;i++) read(q[i].l), read(q[i].r), read(q[i].k1), read(q[i].k2), q[i].pos=i; sort(q+1, q+1+m); for(int i=1, l=1, r=0;i<=m;i++) { while(l<q[i].l) update(l++, -1); while(l>q[i].l) update(--l, 1); while(r<q[i].r) update(++r, 1); while(r>q[i].r) update(r--, -1); x=query1(q[i].k1); ans[q[i].pos]=v[x][query2(x, q[i].k2)]; } for(int i=1;i<=m;i++) printf("%d\n", ans[i]); }
View Code

bzoj3920: Yuuna的禮物(莫隊+分塊套分塊)