2020-2021年度第二届全国大学生算法设计与编程挑战赛 (春季赛)D zeal题题解

题目大意:给你n个数与q个询问
每个区间给你一个l,r与k
对于每个循环回答区间[l,r]中出现k次数字有多少种
思路:莫队算法
先将数组分为√n块
对于每个询问进行左端点所处的块从小到大排序操作(若左端点相等进行右端点所处的块进行从小到大排序操作)
之后利用双指针进行添加删除元素操作得出询问答案即可
时间复杂度是O(n√n)
因为询问的时候可能第一个询问会出现左端点在第一个块,右端点在第一个块
第二个询问左端点在第一个块,右端点在最后一个块的情况
那么这个时候需要大约移动n次左右指针才能完成左边一个块询问的情况
而一共又因为一共有√n个块,所以是O(n√n)的时间复杂度

AC代码:

#include 
#include 
#include 
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;
}

你可能感兴趣的:(题解,算法)