[SDOI2017]序列计数

一、题目

点此看题

二、解法

0x01 朴素dp
需要把题目中的这n个数的和是p的倍数转化成 这$n$个数的和$\%p$等于$0$，因为保证至少一个数是质数很难，我们可以用 全部-不含质数的方案数，设$dp[i][j]$为前$i$个数的和$\%p$的值为$j$的总方案数，转移如下：
$$dp[i][j]=\sum dp[i-1][(j-k+p)\%p]\times cnt[k]$$时间复杂度$O(n{p}^2)$，可以拿到$30$分的高分。

0x02 矩阵加速
看到$n$这么大应该能想到矩阵加速把，我们可以构造出如下的矩阵：

用来乘的矩阵：
$\begin{array}{c}cn{t}_0\\ cn{t}_1\\ ......\\ cn{t}_{p-1}\end{array}$

快速幂矩阵：
$\begin{array}{ccc}cn{t}_0& cn{t}_{p-1}& cn{t}_{p-2}......\\ cn{t}_1& cn{t}_0& cn{t}_{p-1}......\\ .......& & \end{array}$
其实构造方法就是让两个矩阵对应元素和一定，然后我们跑两遍快速幂。（下面还有一种方法）

#include <cstdio>
#include <cstring>
const int mod = 20170408;
const int MAXM = 2e7+5; 
int read()
{
    int num=0,flag=1;char c;
    while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;
    while(c>='0'&&c<='9')num=(num<<3)+(num<<1)+(c^48),c=getchar();
    return num*flag;
}
int n,m,p,tot,ans[2],cnt[105],pr[MAXM/10];
bool vis[MAXM];
struct Matrix
{
	int n,m,a[105][105];
	Matrix() {n=m=0;memset(a,0,sizeof a);}
	Matrix operator * (const Matrix &x) const
    {
        Matrix r;
        for(int i=1;i<=n;i++)
            for(int j=1;j<=m;j++)
                for(int k=1;k<=x.m;k++)
                    r.a[i][k]=(r.a[i][k]+1ll*a[i][j]*x.a[j][k])%mod;
        r.n=n;r.m=x.m;
        return r;
    }
}A,F;
Matrix qkpow(Matrix a,int b)
{
	Matrix res;
	res.n=res.m=p;
	for(int i=1;i<=p;i++)
		res.a[i][i]=1;
	while(b>0)
	{
		if(b&1) res=res*a;
		a=a*a;
		b>>=1;
	}
	return res;
}
void fuck(int f)
{
	A.n=A.m=p;
	F.n=p;F.m=1;
	for(int i=1;i<=p;i++)
		for(int j=1;j<=p;j++)
			A.a[i][j]=cnt[(i-j+p)%p];
	for(int i=1;i<=p;i++)
		F.a[i][1]=cnt[i-1];
	ans[f]=(qkpow(A,n-1)*F).a[1][1];
}
signed main()
{
	n=read();m=read();p=read();
	for(int i=0;i<p;i++) cnt[i]=m/p;
	for(int i=1;i<=m%p;i++)
		cnt[i]++;
	fuck(0);
	for(int i=2;i<=m;i++)
	{
		if(!vis[i]) cnt[i%p]--,pr[++tot]=i;
		for(int j=1;j<=tot && i*pr[j]<=m;j++)
		{
			vis[i*pr[j]]=1;
			if(i%pr[j]==0) break;
		}
	}
	fuck(1);
	printf("%d\n",(ans[0]-ans[1]+mod)%mod);
}

0x03 对dp深入思考
发现本题选数的限制很小，我们可以改写$dp$方程，设$f[i][j]$为有$i$个数$\%p$后的值为$j$的方案数，转移：
$$f[i][j]=\sum f[mid][k]\times f[mid][(j-k+p)\%p]$$现在我们就只需要求$mid$的$dp$值了，然后不需要优化的时间复杂度是$O(\log n{p}^2)$。
这个式子可以用$\text{FFT}$优化，有优化后的复杂度是$O(\log np\log p)$。

咕咕咕