1. 程式人生 > >【GDOI2018模擬8.12】區間第k小

【GDOI2018模擬8.12】區間第k小

Description

給出一個長度為n的序列a,q次詢問某個區間[l,r]中的區間第k小,注意如果一個數的出現次數大於w就把它當成n
詢問強制線上
n,q,ai<=10^5

Solution

Orz 資料結構 根號演算法講師
首先如果詢問可以離線怎麼做?
一個顯然的思路就是莫隊+資料結構直接做,但是這樣是O(nnlogn)
實測資料無梯度一個點都跑不過QwQ
但是我們觀察莫隊,我們最壞可能有NN次插入,但只有N次詢問,我們考慮平衡一下這兩個複雜度
觀察到值域也只有n,所以我們可以同時對值域分塊
我們維護Fi表示當前區間中值在值域第i塊的出現次數<=w的數有多少個,再維護一個Di表示i這個數出現了多少次,那麼就可以做到O(1)插入或刪除
然後對於每次詢問我們暴力掃描所有塊,找到答案屬於哪個塊中,然後再暴力進入這個塊中查詢
這樣就做到總複雜度O(N

N)了
但是這道題強制線上怎麼辦?
我們直接分塊,維護Fi,j,k表示第i塊到第j塊中值在值域第k塊出現次數<=w的數有多少個,這個可以O(NN)預處理
再維護一個gi,j表示前i塊數j出現了多少次,這個也可以O(NN)預處理
那麼對於每次詢問,我們把多餘出來的部分維護一個Di,含義和上文相同,然後把D和F、D和G合併得到和上文功能相同的陣列,按照上文的做法就可以通過此題
總複雜度依舊是O(NN)

Code

#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++) using namespace std; const int N=1e5+5,M=333; int n,q,ty,le,ri,k,u,v,ans,w,sz,l[M+5],r[M+5],a[N],b[N]; int d[N],vis[M+5],g[M+5][N],f[M+5][M+5][M+5]; int read() { char ch; for(ch=getchar();ch<'0'||ch>'9';ch=getchar()); int x=ch-'0'; for(ch=getchar();ch>='0'
&&ch<='9';ch=getchar()) x=x*10+ch-'0'; return x; } int get_ans(int u,int v) { int now=0,sg=0; fo(i,1,sz) { if (now+vis[i]>=k) {sg=i;break;} now+=vis[i]; } if (!sg) return n; int st=l[sg],ed=r[sg]; if (sg==1) st=0; if (sg==sz) ed=n-1; fo(i,st,ed) { if (d[i]<0) continue; if (d[i]+g[v][i]-g[u][i]>w) continue; if (now+d[i]+g[v][i]-g[u][i]>=k) return i; now+=d[i]+g[v][i]-g[u][i]; } } int main() { freopen("kth.in","r",stdin); freopen("kth.out","w",stdout); n=read();w=read();q=read();ty=read(); fo(i,1,n) a[i]=read(),b[i]=(i-1)/M+1; sz=(n-1)/M+1;b[0]=1; fo(i,1,sz) l[i]=r[i-1]+1,r[i]=min(n,l[i]+M-1); fo(i,1,sz) { fo(j,0,n-1) g[i][j]=g[i-1][j]; fo(j,l[i],r[i]) g[i][a[j]]++; } fo(i,1,sz) { fo(j,0,n-1) d[j]=0; fo(j,i,sz) { fo(k,1,sz) f[i][j][k]=f[i][j-1][k]; fo(p,l[j],r[j]) { if (d[a[p]]<0) continue; d[a[p]]++; if (d[a[p]]>w) f[i][j][b[a[p]]]-=w,d[a[p]]=-1; else f[i][j][b[a[p]]]++; } } } fo(i,0,n-1) d[i]=0; for(;q;q--) { le=read();ri=read();k=read(); if (ty) le^=ans,ri^=ans,k^=ans; fo(i,1,sz) if (l[i]<=le&&r[i]>=le) {u=i;break;} fo(i,1,sz) if (l[i]<=ri&&r[i]>=ri) {v=i;break;} if (u+1>=v) { fo(i,1,sz) vis[i]=0; fo(i,le,ri) { if (d[a[i]]<0) continue; d[a[i]]++; if (d[a[i]]>w) vis[b[a[i]]]-=w,d[a[i]]=-1; else vis[b[a[i]]]++; } ans=get_ans(0,0); printf("%d\n",ans); fo(i,le,ri) d[a[i]]=0; continue; } fo(i,1,sz) vis[i]=f[u+1][v-1][i]; fo(i,le,r[u]) { if (d[a[i]]<0) continue; if (g[v-1][a[i]]-g[u][a[i]]>w) continue; d[a[i]]++; if (d[a[i]]+g[v-1][a[i]]-g[u][a[i]]>w) vis[b[a[i]]]-=w,d[a[i]]=-1; else vis[b[a[i]]]++; } fo(i,l[v],ri) { if (d[a[i]]<0) continue; if (g[v-1][a[i]]-g[u][a[i]]>w) continue; d[a[i]]++; if (d[a[i]]+g[v-1][a[i]]-g[u][a[i]]>w) vis[b[a[i]]]-=w,d[a[i]]=-1; else vis[b[a[i]]]++; } ans=get_ans(u,v-1); fo(i,le,r[u]) d[a[i]]=0; fo(i,l[v],ri) d[a[i]]=0; printf("%d\n",ans); } }