[SDOI2017]序列计数

最新推荐文章于 2022-02-09 10:41:49 发布

OneInDark

最新推荐文章于 2022-02-09 10:41:49 发布

阅读量93

点赞数

分类专栏： # 多项式/生成函数动态规划 C++

本文链接：https://blog.csdn.net/qq_42101694/article/details/103588283

版权

C++ 同时被 3 个专栏收录

547 篇文章 3 订阅

订阅专栏

动态规划

220 篇文章 2 订阅

订阅专栏

多项式/生成函数

51 篇文章 1 订阅

订阅专栏

题目

传送门 to luogu

思路

感谢@dyh1654733675 的博客提供了思路。~~和代码。~~

基本思路，是从 $\tt{dp}$ 中脱胎换骨：

如果用 $f (i, j)$ 表示 $i$ 个数字、余数为 $j$ 的答案，那么对于任意 $L$ 均有 $f(i,j)=\sum_{x=0}^{p-1}f(L,x)\times f(i-L,(j+p-x)\bmod p)$

也就是枚举前 $L$ 个数字的余数，与剩下的余数相加。

然后我们写一个函数 $g(L,x)=\sum_{i=0}^{p-1}f(L,i)x^i$ ，因为 $\tt{dp}$ 转移非常像卷积。

于是我们就可以写成快速幂的形式。求的答案就是 $g^n(1,x)$ ~~中的某一项。~~

嗯？你问我怎么处理 “有一个质数” ？正难则反，减去全部是合数的情况即可。

代码

#include<iostream>
#include<cstdio>
#define mod 20170408
using namespace std;
int n,m,p,vis[20000010],cnt,pre[2000000];
long long f[210],F[210],g[210],G[210];
long long c[210],x=1;
void work(long long *a,long long *b,long long *d) {
    int i,j;
    for(i=0; i<p; i++)
        for(j=0; j<p; j++) c[i+j]+=a[i]*b[j];
    for(i=0; i<p; i++) {
    d[i]=(c[i]+c[i+p])%mod;
    c[i]=c[i+p]=0;
    }return ;
}
int main() {
    int i,j; 
    cin>>n>>m>>p;
    F[0]=G[0]=f[1]=g[1]=1;
    for(i=2; i<=m; i++) {
        f[i%p]++;       //此时的f为f[1][0--p]的值
        if(!vis[i])pre[++cnt]=i;
        else g[i%p]++;    //通过欧拉筛记录g[1][0--p]的值
        for(j=1; pre[j]*i<=m; j++) {
            vis[pre[j]*i]=1;
            if(!(i%pre[j])) break;
        }
    }
    while(n) {
        if(n&1)
            work(F,f,F),work(G,g,G);
        work(f,f,f);
        work(g,g,g);
        n>>=1;
    }
    cout<<(F[0]-G[0]+mod)%mod;  //特别注意这里，学长特别提醒我了的。F[0]可能比G[0]小。

    return 0;
}