题目大意: 有多少个长度为 n n n 的序列满足:每个数字都在 [ 1 , m ] [1,m] [1,m] 内;序列总和为 p p p 的倍数;至少有一个质数。
题解
至少有一个质数的方案数
容易转化成无限制的方案数
-一个质数都没有的方案数
。
令 f i f_i fi 表示可以使用的数字中,模 p p p 余 i i i 的数字个数,那么 [ x 0 ] f n [x^0]f^n [x0]fn 就是序列总和为 p p p 的倍数的方案数,这个可以用快速幂,每次暴力卷积来实现。
代码如下:
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define maxn 20000010
#define mod 20170408
int n,m,p;
int prime[maxn],t=0;
bool v[maxn];
void SieveInit(){
v[1]=true;
for(int i=2;i<=m;i++){
if(!v[i])prime[++t]=i;
for(int j=1;j<=t&&i*prime[j]<=m;j++){
v[i*prime[j]]=true;
if(i%prime[j]==0)break;
}
}
}
int f[110],g[110],tmp[110],ans=0;
void add(int &x,int y){x=(x+y>=mod?x+y-mod:x+y);}
void Mul(int *a,int *b){
memset(tmp,0,sizeof(tmp));
for(int i=0;i<p;i++){
for(int j=0;j<p;j++){
add(tmp[(i+j)%p],1ll*a[i]*b[j]%mod);
}
}
for(int i=0;i<p;i++)a[i]=tmp[i];
}
void ksm(){
memset(g,0,sizeof(g));g[0]=1;
int y=n;while(y){
if(y&1)Mul(g,f);
y>>=1;Mul(f,f);
}
}
int main()
{
scanf("%d %d %d",&n,&m,&p);SieveInit();
for(int i=1;i<=m;i++)f[i%p]++;
ksm();ans=g[0];
memset(f,0,sizeof(f));
for(int i=1;i<=m;i++)if(v[i])f[i%p]++;
ksm();ans=(ans-g[0]+mod)%mod;
printf("%d",ans);
}