51nod 1230：幸运数

51nod 1230：幸运数

题目链接：http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1230

题目大意：如果一个数各个数位上的数字之和是质数，并且各个数位上的数字的平方和也是质数，则称它为幸运数。例如：120是幸运数，因为120的数字之和为3，平方和为5，均为质数，所以120是一个幸运数字。给定x，y，求x，y之间（ 包含x，y，即闭区间[x,y]）有多少个幸运数。

数位DP

代码如下：

#include <cstdio>
#include <cstring>
using namespace std;
typedef long long ll;
const int maxn=22;
int dig[maxn],prime[1459]={0},num_prime=0,T;
ll f[maxn][163][1459],x,y;
bool p[1459]={1,1};
ll dfs(int pos,int sum,int sqrsum,int limit){
    if (pos<0) return (!p[sum])&&(!p[sqrsum]);
    if (!limit&&f[pos][sum][sqrsum]!=-1) return f[pos][sum][sqrsum];
    ll res=0;
    int last=limit?dig[pos]:9;
    for (int i=0;i<=last;i++){
        res+=dfs(pos-1,sum+i,sqrsum+i*i,limit&&(i==last));
    }
    if (!limit) f[pos][sum][sqrsum]=res;
    return res;
}
ll solve(ll n){
    int len=0;
    while (n){
        dig[len++]=n%10;
        n/=10;
    }
    return dfs(len-1,0,0,1);
}
void init(){
    memset(f,-1,sizeof(f));
    for(long i=2;i<1459;i++){
        if(!p[i])prime[num_prime++]=i;
        for(long j=0;j<num_prime&&i*prime[j]<1459;j++){
            p[i*prime[j]]=1;
            if(!(i%prime[j]))break;
        }
    }
}
int main(void){
    init();
    scanf("%d",&T);
    while(T--){
        scanf("%lld%lld",&x,&y);
        printf("%lld\n",solve(y)-solve(x-1));
    }
}