[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]);
}

注意!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系我们删除。



 
粤ICP备14056181号  © 2014-2021 ITdaan.com