题目大意:给你n个数与q个询问
每个区间给你一个l,r与k
对于每个循环回答区间[l,r]中出现k次数字有多少种
思路:莫队算法
先将数组分为√n块
对于每个询问进行左端点所处的块从小到大排序操作(若左端点相等进行右端点所处的块进行从小到大排序操作)
之后利用双指针进行添加删除元素操作得出询问答案即可
时间复杂度是O(n√n)
因为询问的时候可能第一个询问会出现左端点在第一个块,右端点在第一个块
第二个询问左端点在第一个块,右端点在最后一个块的情况
那么这个时候需要大约移动n次左右指针才能完成左边一个块询问的情况
而一共又因为一共有√n个块,所以是O(n√n)的时间复杂度
AC代码:
#include <iostream>
#include <math.h>
#include <algorithm>
using namespace std;
struct node
{
int l,r,id;
int k;
};
node q[40010];
int b[40010];
int pos[40010];
int ans[40010];
int sum1[40010];
int sum2[40010];//存有多少种相同的出现次数
void add(int x)
{
sum2[sum1[x]]--;//现在x出现次数加1了,之前存的出现次数要减去一种元素
sum1[x]++;
sum2[sum1[x]]++;
return;
}
void del(int x)
{
sum2[sum1[x]]--;//现在x出现次数加1了,之前存的出现次数要减去一种元素
sum1[x]--;
sum2[sum1[x]]++;
return;
}
int main()
{
int n,m;
cin>>n>>m;
int siz=sqrt(n);
for(int i=1; i<=n; i++)
{
cin>>b[i];
pos[i]=i/siz;//分块
}
for(int i=0; i<m; i++)
{
cin>>q[i].l>>q[i].r>>q[i].k;
q[i].id=i;
}
sort(q,q+m,[](node x,node y)
{
return pos[x.l]==pos[y.l]?x.r<y.r:x.l<y.l;//按左端点所在的块进行排序,如果一样就看右端点
});
int l=1,r=0;
for(int i=0; i<m; i++)
{
while(r<q[i].r)add(b[++r]);
while(l>q[i].l)add(b[--l]);
while(l<q[i].l)del(b[l++]);
while(r>q[i].r)del(b[r--]);
ans[q[i].id]=sum2[q[i].k];
}
for(int i=0; i<m; i++)cout<<ans[i]<<endl;
return 0;
}