P3730 曼哈顿交易 [莫队]

传送门

离散化之后 , 莫队维护每个数字出现的次数

对于次数,我们继续分块维护 , 这样是O(1)修改 , O(根号n)查询

复杂度 

#include<bits/stdc++.h>
#define N 100050
using namespace std;
struct Node{int l,r,k,id;}q[N];
int a[N],b[N],n,m,pos[N],ans[N];
int cnt[N],L[N],R[N],num[N],sum[N],tot,ret;
bool cmp(Node a,Node b){
	if(pos[a.l]==pos[b.l]) return a.r<b.r;
	return pos[a.l]<pos[b.l];
}
int read(){
	int cnt=0;char ch=0;
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))cnt=cnt*10+(ch-'0'),ch=getchar();
	return cnt;
}
void Modify(int x,int w){num[x]+=w , sum[pos[x]]+=w , tot+=w;}
void del(int x){ Modify(cnt[x],-1); cnt[x]--; if(cnt[x]) Modify(cnt[x],1);}
void add(int x){ if(cnt[x]) Modify(cnt[x],-1); cnt[x]++; Modify(cnt[x],1);}
int Q(int k){
	if(tot<k) return -1;
	int now=1; 
	while(k-sum[now]>0) k-=sum[now++];
	for(int i=L[now];i<=R[now];i++){
		k-=num[i]; if(k<=0) return i;
	}
}
int main(){
	n=read(),m=read();
	for(int i=1;i<=n;i++) a[i]=b[i]=read();
	sort(b+1,b+n+1); int siz=unique(b+1,b+n+1)-(b+1);
	for(int i=1;i<=n;i++) a[i]=lower_bound(b+1,b+siz+1,a[i])-b;
	
	siz = sqrt(n);
	for(int i=1;i<=n;i++) pos[i]=(i-1)/siz+1;
	for(int i=1;i<=n;i+=siz) L[++ret]=i,R[ret]=i+siz-1;R[ret]=n;
	
	for(int i=1;i<=m;i++){
		q[i].l=read(),q[i].r=read(),q[i].k=read(),q[i].id=i;
	}
	sort(q+1,q+m+1,cmp);
	int l,r; l=r=1; add(a[1]);
	for(int i=1;i<=m;i++){
		while(r>q[i].r) del(a[r--]); while(r<q[i].r) add(a[++r]);
		while(l<q[i].l) del(a[l++]); while(l>q[i].l) add(a[--l]);
		ans[q[i].id] = Q(q[i].k); 
	}
	for(int i=1;i<=m;i++) printf("%d\n",ans[i]); return 0;
}