HDOJ 3652 B-number

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=3652

题意：给我们一个数n，问我们从1~n有多少数既含有13又是13的倍数。

这道题和HDOJ 3555 Bomb很像，但是多了一种状态的判定，那就是取模。本题小编还是采取两种思路进行求解

1.DP:

dp[i][j][0]表示的是在长度为i的情况下，取模为j时，不含13的个数。

dp[i][j][1]表示的是在长度为i的情况下，取模为j时，不含13，但首位为3的个数。

dp[i][j][2]表示的是在长度为i的情况下，取模为j时，含13的个数。

则有

dp[i+1][temp_j][0] += dp[i][j][0];
dp[i+1][temp_j][2] += dp[i][j][2];

当最高位为3时：dp[i+1][temp_j][1] += dp[i][j][0];

当最高位为1时：dp[i+1][temp_j][0] -= dp[i][j][1];
                           dp[i+1][temp_j][2] += dp[i][j][1];

那么我们从高到低枚举每一位，当我们枚举的这位是1的话，ans += dp[i-1][temp_mod][1];当我们枚举到之前有13的话，ans += dp[i-1][temp_mod][0];

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 10;
const int MOD = 13;
int dp[maxn][MOD][3], n;
int mod[maxn];
void init()
{
	mod[1] = 1;
	for(int i=2; i<maxn; i++)
		mod[i] = (mod[i-1]*10) % MOD;
	dp[0][0][0] = 1;
	for(int j=0; j<=9; j++)
		dp[1][j][0] = 1;
	dp[1][3][1] = 1;
	for(int i=1; i<maxn-1; i++)
		for(int j=0; j<MOD; j++)
			for(int k=0; k<=9; k++)
			{
				int temp_j = (j+k*mod[i+1]) %MOD;
				dp[i+1][temp_j][0] += dp[i][j][0];
				dp[i+1][temp_j][2] += dp[i][j][2];
				if(k == 1)
				{
					dp[i+1][temp_j][0] -= dp[i][j][1];
					dp[i+1][temp_j][2] += dp[i][j][1];
				}
				if(k == 3)
					dp[i+1][temp_j][1] += dp[i][j][0];
			}
}
int solve(int a)
{
	int bit[maxn];
	int ans = 0,len = 0;
	while(a)
	{
		bit[++len] = a%10;
		a /= 10;
	}
	bool flag = false;
	int pre = 0;
	for(int i=len; i>=1; i--)
	{
		for(int j=0; j<bit[i]; j++)
		{
			int temp_mod = (13-(pre*10 + j) * mod[i]%MOD) % MOD;
			ans += dp[i-1][temp_mod][2];
			if(flag || j==3 && pre%10 == 1)
				ans += dp[i-1][temp_mod][0];
			else if(j == 1)
				ans += dp[i-1][temp_mod][1];
		}
		pre = (pre*10) + bit[i];
		if(i != len && bit[i+1] == 1 && bit[i] == 3)
			flag = true;
	}
	return ans;
}
int main()
{
	init();
	while(scanf("%d", &n) !=EOF)
		printf("%d\n", solve(n+1));
}

2.DFS 记忆化搜索

我们用dp[i][j][k]来分别记录状态，表示枚举到第I位的时候，对13取模为j时，是否还有13(2/0)，或者不含13且最后一位是1的情况的个数。

#include <cstdio>//DFS
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 16;
int bit[maxn];
int dp[maxn][maxn][3];
int dfs(int len, int mod, int have, int ismax)
{
	if(len <= 0) return mod==0 && have==2;
	if(!ismax && dp[len][mod][have]!=-1) return dp[len][mod][have];
	int Max = ismax? bit[len]:9;
	int ans = 0;
	for(int i=0; i<=Max; i++)
	{
		if(have==2) ans += dfs(len-1, (mod*10+i)%13, 2, ismax&&i==Max);
		else if(i == 1) ans += dfs(len-1, (mod*10+i)%13, 1, ismax&&i==Max);
		else if(i==3 && have==1) ans += dfs(len-1, (mod*10+i)%13, 2, ismax&&i==Max);
		else ans += dfs(len-1, (mod*10+i)%13, 0, ismax&&i==Max);
	}
	return ismax? ans : dp[len][mod][have]=ans;
}
int solve(int n)
{
	memset(dp,-1,sizeof(dp));
	int len = 0;
	while(n)
	{
		bit[++len] = n%10;
		n /= 10;
	}
	return dfs(len, 0, 0, 1);
}
int main()
{
	int n;
	while(scanf("%d", &n) !=EOF)
		printf("%d\n", solve(n));
}