0
点赞
收藏
分享

微信扫一扫

洛谷P2359 三素数数【DP】【绿】

玉字璧 2022-03-27 阅读 37

Date:2022.03.27
题目描述
如果一个数的所有连续三位数字都是大于100的素数,则该数称为三素数数。比如113797是一个6位的三素数数。
输入格式
一个整数n(3 ≤ n ≤ 10000),表示三素数数的位数。
输出格式
一个整数,表示n位三素数的个数m,要求输出m除以10^9 + 9的余数。
输入输出样例
输入 #1复制
4
输出 #1复制
204

思路:开始讨论了最后加上1、2、3位数,发现只需要讨论最后1位数即可非常方便实现状态转移。并且不论结尾加几位数都与加1位数重叠。
由此,状态转移方程: f [ i ] [ j ] [ k ] : 共 i 位 的 数 , 末 尾 两 位 分 别 为 j 、 k 的 三 素 数 数 的 个 数 。 f[i][j][k]:共i位的数,末尾两位分别为j、k的三素数数的个数。 f[i][j][k]:ijk
状态转移: f [ i ] [ j ] [ k ] = f [ i + 1 ] [ k ] [ u ] + f [ i ] [ j ] [ k ] ; 【 前 提 保 证 j ∗ 100 + k ∗ 10 + u 是 素 数 & & 要 > 100 】 f[i][j][k]=f[i+1][k][u]+f[i][j][k];【前提保证j*100+k*10+u是素数\&\&要>100】 f[i][j][k]=f[i+1][k][u]+f[i][j][k];j100+k10+u&&>100
初始状态: f [ 2 ] [ j ] [ k ] = 1 ; 【 k ∈ [ 0 − 9 ] , j ∈ [ 0 − 9 ] 】 f[2][j][k]=1;【k\in[0-9],j\in[0-9]】 f[2][j][k]=1;k[09],j[09]
结果: f [ n ] [ i ] [ j ] 求 和 【 i ∈ [ 0 − 9 ] , j ∈ [ 0   9 ] 】 f[n][i][j]求和【i\in[0-9],j\in[0~9]】 f[n][i][j]i[09],j[0 9]

代码如下:

#include <bits/stdc++.h>
#define x first
#define y second
using namespace std;
typedef long long LL;
typedef pair<LL, LL> PII;
const LL N = 10100,INF=0x3f3f3f3f3f3f3f3f,mod=1e9+9;
typedef long long LL;
LL n,m,f[N][11][11];
bool st[1010];
bool is_prime(LL x)
{
    for(int i=2;i<x;i++)
        if(x%i==0) return false;
    return true;
}
int main()
{
    ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    cin>>n;
    for(int i=2;i<=1000;i++)
        if(is_prime(i)) st[i]=true;
    for(int j=0;j<=9;j++)
        for(int k=0;k<=9;k++) f[2][j][k]=1;
    for(int i=2;i<=n;i++)
        for(int j=0;j<=9;j++)
            for(int k=0;k<=9;k++)
                for(int u=0;u<=9;u++)
                {
                    LL x=j*100+k*10+u;
                    if(x>100&&st[x]) f[i+1][k][u]=(f[i+1][k][u]+f[i][j][k])%mod;
                }
    LL ans=0;
    for(int i=1;i<=9;i++)
        for(int j=1;j<=9;j++)
            ans=(ans+f[n][i][j])%mod;
    cout<<ans;
    return 0;
}
举报

相关推荐

0 条评论