SDOI 2017 序列计数 题解

题目传送门

题目大意： 有多少个长度为 $n$ 的序列满足：每个数字都在 $[1,m]$ 内；序列总和为 $p$ 的倍数；至少有一个质数。

题解

至少有一个质数的方案数容易转化成无限制的方案数-一个质数都没有的方案数。

令 $f_i$ 表示可以使用的数字中，模 $p$ 余 $i$ 的数字个数，那么 $[x^0]f^n$ 就是序列总和为 $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);
}