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

杭电暑假集训 Team Contests - Warmup G题_第1张图片

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

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

#include 
#include 
#include 
#include 
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;
}


你可能感兴趣的:(ACM_数位DP)