首页 > 代码库 > ural Threeprime Numbers(dp)
ural Threeprime Numbers(dp)
http://acm.timus.ru/problem.aspx?space=1&num=1586
题意没看懂,看了别人的翻译。threeprime number的意思是任意三个连续的数组成的一个三位数是素数,注意必须是三位数。给出n,问满足条件的n位数有多少个。
先把三位数的素数筛选出来并标记,设dp[i][j][k]表示到i位为止,最后两位是j和k的满足条件的数的个数。
那么dp[i][j][k] += dp[i-1][g][j]。
#include <stdio.h> #include <iostream> #include <map> #include <set> #include <list> #include <stack> #include <vector> #include <math.h> #include <string.h> #include <queue> #include <string> #include <stdlib.h> #include <algorithm> #define LL __int64 #define eps 1e-12 #define PI acos(-1.0) #define PP pair<LL,LL> using namespace std; const int INF = 0x3f3f3f3f; const int maxn = 1000; const int mod = 1000000009; int prime[maxn+10]; int flag[maxn+10]; int cnt = 0; int tprime[maxn+10]; int dp[10010][12][12]; void get_prime() { memset(flag,0,sizeof(flag)); prime[0] = 0; flag[0] = flag[1] = 1; for(int i = 2; i <= maxn; i++) { if(flag[i] == 0) { prime[++prime[0]] = i; if(i >= 100 && i <= 999) tprime[++cnt] = i; } for(int j = 1; j <= prime[0] && i*prime[j] <= maxn; j++) { flag[prime[j]*i] = 1; if(i % prime[j] == 0) break; } } } int main() { get_prime(); int n; while(~scanf("%d",&n)) { if(n == 3) { printf("%d\n",cnt); continue; } memset(dp,0,sizeof(dp)); for(int i = 1; i <= cnt; i++) { int b = tprime[i]%10; int a = tprime[i]/10%10; dp[3][a][b] += 1; } for(int i = 4; i <= n; i++) { for(int j = 0; j <= 9; j++) { for(int k = 0; k <= 9; k++) { for(int g = 0; g <= 9; g++) { int num = j*100 + k*10 + g; if(flag[num] == 0 && num >= 100) //num必须是三位数 { dp[i][k][g] += dp[i-1][j][k]%mod; dp[i][k][g] %= mod; } } } } } LL ans = 0; for(int i = 0; i <= 9; i++) { for(int j = 0; j <= 9; j++) ans = (ans + dp[n][i][j])%mod; } printf("%I64d\n",ans); } return 0; }
ural Threeprime Numbers(dp)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。