[CodeM初赛A轮]E

题解

大小点分治。
出现次数大于阈值的，处理前缀和，然后暴力枚举一个询问判断。
出现次数小于阈值的，跑莫队，维护每个数出现次数，以及每种出现次数的数的个数，然后询问可以暴力枚举出现次数。

#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
const int maxn=50000+10,B=240;
struct dong{
    int l,r,k,id;
} ask[maxn];
int a[maxn],sum[maxn],belong[maxn],cnt[maxn],c[maxn],d[maxn];
int ans[maxn];
int i,j,k,l,r,t,n,m,tot,top;
int read(){
    int x=0,f=1;
    char ch=getchar();
    while (ch<'0'||ch>'9'){
        if (ch=='-') f=-1;
        ch=getchar();
    }
    while (ch>='0'&&ch<='9'){
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
int gcd(int a,int b){
    return b?gcd(b,a%b):a;
}
bool cmp(dong a,dong b){
    return belong[a.l]<belong[b.l]||belong[a.l]==belong[b.l]&&a.r<b.r;
}
void ins(int x){
    if (cnt[a[x]]>B) return;
    if (c[a[x]]) d[c[a[x]]]--;
    c[a[x]]++;
    if (c[a[x]]) d[c[a[x]]]++;
}
void del(int x){
    if (cnt[a[x]]>B) return;
    if (c[a[x]]) d[c[a[x]]]--;
    c[a[x]]--;
    if (c[a[x]]) d[c[a[x]]]++;
}
int main(){
    n=read();m=read();
    fo(i,1,n) a[i]=read(),cnt[a[i]]++,belong[i]=(i-1)/B+1;
    fo(i,1,m) ask[i].l=read(),ask[i].r=read(),ask[i].k=read(),ask[i].id=i;
    fo(i,1,n)
        if (cnt[i]>B){
            fo(j,1,n) sum[j]=sum[j-1]+(a[j]==i);
            fo(j,1,m){
                if (sum[ask[j].r]-sum[ask[j].l-1]>0&&gcd(sum[ask[j].r]-sum[ask[j].l-1],ask[j].k)==1) ans[ask[j].id]++;
            }
        }
    sort(ask+1,ask+m+1,cmp);
    l=1;r=0;
    fo(i,1,m){
        while (l<ask[i].l) del(l++);
        while (l>ask[i].l) ins(--l);
        while (r<ask[i].r) ins(++r);
        while (r>ask[i].r) del(r--);
        fo(j,1,B)
            if (gcd(j,ask[i].k)==1) ans[ask[i].id]+=d[j];
    }
    fo(i,1,m) printf("%d\n",ans[i]);
}