NC17942J(莫队算法)

给出一个序列,每次询问区间内出现次数恰好是k的元素个数。

//c[maxn]维护出现i次的数有多少
//cc[maxn]维护第i个数出现多少次
//然后莫队
#include<bits/stdc++.h>
using namespace std;
const int maxn=4e4+100;
int c[maxn],cc[maxn];
int belong[maxn];
int sz,bnum;
struct node {
	int l,r,k,id;
	bool operator < (const node &b) const {
		return  (belong[l]^belong[b.l])?belong[l]<belong[b.l]:((belong[l]&1)?r<b.r:r>b.r);
	}
}q[maxn];
int n,m,a[maxn],ans[maxn];
int main () {
	scanf("%d%d",&n,&m);
	for (int i=1;i<=n;i++) scanf("%d",a+i),cc[a[i]]++;
	int l=1,r=n;
	sz=sqrt(n);
	bnum=n/sz+(n%sz>0);
	for (int i=1;i<=bnum;i++) {
		for (int j=(i-1)*sz+1;j<=i*sz;j++) {
			belong[j]=i;
		}
	}
	for (int i=0;i<=4e4;i++) {
		c[cc[i]]++;
	}
	for (int i=1;i<=m;i++) {
		scanf("%d%d%d",&q[i].l,&q[i].r,&q[i].k);
		q[i].id=i;
	}
	sort(q+1,q+m+1);
	for (int i=1;i<=m;i++) {
		while (l<q[i].l) {
			c[cc[a[l]]]--;
			cc[a[l]]--;
			c[cc[a[l]]]++;
			l++;
		}
		while (l>q[i].l) {
			l--;
			c[cc[a[l]]]--;
			cc[a[l]]++;
			c[cc[a[l]]]++;
		}
		while (r<q[i].r) {
			r++;
			c[cc[a[r]]]--;
			cc[a[r]]++;
			c[cc[a[r]]]++;
		}
		while (r>q[i].r) {
			c[cc[a[r]]]--;
			cc[a[r]]--;
			c[cc[a[r]]]++;
			r--;
		}
		ans[q[i].id]=c[q[i].k];
	}
	for (int i=1;i<=m;i++) printf("%d\n",ans[i]);
}

posted @ 2021-07-15 18:48  zlc0405  阅读(46)  评论(0编辑  收藏  举报