HDU XXXX：求[L,R]的素数数量（数位DP）

Problem G

Time Limit : 2000/1000ms (Java/Other)   Memory Limit : 131072/131072K (Java/Other)

Total Submission(s) : 62   Accepted Submission(s) : 28

Font: Times New Roman | Verdana | Georgia

Font Size: ← →

Problem Description

S number is the number which the sum of every digit is a prime number, such as number 98, 29. Output the number of S number in [L,R].

Input

  First line contains T(T≤10) denoting the number of test cases.
  T cases follows For each cases: 
  There two numbers L,R.(0≤L≤R≤1016)

Output

For each case, output the number of S number.

Sample Input

2
4 30
49 173

Sample Output

12
45

题意：找出[L,R]里面素数的总和，即求出[0,R]-[0,L-1]就可以了。
思路：因为数据范围在0~10^16那么大，所以不可以暴力了，考虑到每个位最多是9，那么最多15个9的话就是135个数那么多，因此我打了个判断135里面的数哪个是素数的表，然后就数位DP

 #include <cstdio>
 #include <iostream>
 #include <cstring>
 using namespace std;
 #define N 20
 int bit[N];
 long long dp[N][][];
 int prime[];

 bool check(int x)
 {
     for(int i=;i*i<=x;i++){
         if(x%i==) return false;
     }
     return x!=;
 }

 void P()
 {
     for(int i=;i<=;i++){
         if(check(i)){
              prime[i]=;
         }
     }
 }
 //flag表示之前的数是否是上界的前缀（即后面的数能否任意填）。
 //flag为 1 表示之前的数不是前缀，可以任意填
 long long dfs(int pos,int st,int have,int flag)
 {
     if(!pos) return have;
     if(flag&&dp[pos][st][have]!=-) return dp[pos][st][have];
     long long ans=;
     int u=flag?:bit[pos];
     for(int d=;d<=u;d++){
         ans+=dfs(pos-,st+d,prime[st+d],!flag&&d==u);
         //判断之前位置的和加上当前位置是否可以是一个素数
     }
     if(flag) dp[pos][st][have]=ans;
     return ans;
 }

 long long solve(long long s)
 {
     memset(bit,,sizeof(bit));
     int l=;
     while(s){
         bit[++l]=s%;
         s/=;
     }
     return dfs(l,,,);
 }

 int main()
 {
     int t;
     cin>>t;
     memset(prime,,sizeof(prime));
     P();
     while(t--){
         memset(dp,-,sizeof(dp));
         long long s1,s2;
         cin>>s1>>s2;
 //        cout<<solve(s2)<<" "<<solve(s1-1)<<endl;
         cout<<solve(s2)-solve(s1-)<<endl;
     }
     return ;
 }

。