当前位置: 代码迷 >> 综合 >> 杭电暑假集训 Team Contests - Warmup G题
  详细解决方案

杭电暑假集训 Team Contests - Warmup G题

热度:74   发布时间:2023-12-06 03:19:47.0


来自暑假集训的第一场组队赛,当时很多队伍第一个过的就是这个题目。一看就是一个数位DP,因为恰恰好队伍里面三个人当时都不会数位DP,也是挺纠结的,都过了好长时间,队友才A掉,不过还是蛮佩服队友的,不会数位DP,纯粹凭着想象A掉的,还是挺强的。

我们用dp[i][j]记录枚举到第i位时,和为j的状态,进行DFS记忆化搜索,因为我们总共才16位数,所以每一位的和加起来最多也只能达到144,所以我们直接先预处理1~144之间的质数,用于判断。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 2000+5;
typedef long long LL;
int bit[20];
int prime[40] = {2,3,5,7,11,13,17,19,23,29,31,37,41,43,47,53,59,61,67,71,73,79,83,89,97,101,103,107,109,113,127,131,137,139};
LL dp[20][maxn];
int a[150];
LL dfs(int len, int sum,int ismax)
{if(len == -1) return a[sum];if(!ismax && dp[len][sum]!=-1) return dp[len][sum];LL ans = 0;int Max = ismax? bit[len]:9;for(int i=0; i<=Max; i++)ans += dfs(len-1, sum+i, ismax&&i==Max); return ismax? ans : dp[len][sum]=ans;
}
LL solve(LL n)
{int len = 0;while(n){bit[len++] = n%10;n /= 10;}LL ans = dfs(len-1, 0, 1);return ans;
}
int main()
{int T;scanf("%d", &T);for(int i=0; i<34; i++) a[prime[i]]++;memset(dp,-1,sizeof(dp));while(T--){LL L,R;cin >> L >> R;cout << solve(R)-solve(L-1) << endl;}return 0;
}


  相关解决方案