Codeforces Problem 161E - Polycarpus the Safecracker

先暴力打素数表，然后枚举5位数里（从高位到低位）有i位确定且为x的数有多少个。

 具体解某个数时暴力即可

#include<stdio.h> #include<math.h> #include<vector> #include<string.h> using namespace std; bool vis[100010]; vector<int>dp[5][100010]; int k,l; int a[6][6]; int dfs(int now){ int i,j,x=0,sum=0; for(i=0;i<now;i++) x=x*10+a[now][i]; if(i==l-1) return dp[1][x].size(); for(i=0;i<dp[l-now][x].size();i++){ int tem=dp[l-now][x][i]; for(j=l-1;j>=now;j--){ a[j][now]=a[now][j]=tem%10; tem/=10; } sum+=dfs(now+1); } return sum; } int main(){ int i,j,n; int m=sqrt(100000.5); for(i=2;i<=m;i++) if(!vis[i]){ for(j=i*i;j<=100000;j+=i) vis[j]=1; } for(i=2;i<100000;i++) if(!vis[i]){ int tem=i; tem/=10; for(j=1;j<=4;j++){ //表示5位数有j位个未确定 dp[j][tem].push_back(i); tem/=10; } } scanf("%d",&n); char str[7]; for(j=1;j<=n;j++){ scanf("%s",str); l=strlen(str); for(i=0;str[i]!='\0';i++){ a[0][i]=a[i][0]=str[i]-'0'; } printf("%d\n",dfs(1)); } }