HDU 3555 Bomb(数位DP）

最新推荐文章于 2021-01-22 20:25:28 发布

lab104_yifan

最新推荐文章于 2021-01-22 20:25:28 发布

阅读量573

点赞数

CC 4.0 BY-SA版权

分类专栏：高效算法-动态规划

本文链接：https://blog.youkuaiyun.com/accelerator_/article/details/43131117

高效算法-动态规划专栏收录该内容

147 篇文章

订阅专栏

本文介绍一种数位动态规划(DP)方法，用于解决找出指定范围内所有包含特定字串的数字个数的问题。通过计算不含特定字串的数字数量，再从总数中减去该数量得到最终答案。

找出1 - N中所有包含49字串的数字个数

思路：普通的数位DP，先找出布包含的，然后总数减去不包含的即可

代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

typedef long long ll;
const int N = 25;

int t;
char num[N];
ll dp[N][10], ans;

int main() {
	scanf("%d", &t);
	while (t--) {
		scanf("%s", num);
		int n = strlen(num);
		sscanf(num, "%I64d", &ans);
		ans++;
		int pre = 0, flag = 0;
		memset(dp, 0, sizeof(dp));
		for (int i = 0; i < n; i++) {
			for (int j = 0; j < 10; j++) {
				for (int k = 0; k < 10; k++) {
					if (j == 9 && k == 4) continue;
					dp[i + 1][j] += dp[i][k];
				}
			}
			if (flag) continue;
			for (int j = 0; j < num[i] - '0'; j++) {
				if (j == 9 && pre == 4) continue;
				dp[i + 1][j]++;
			}
			if (pre == 4 && num[i] - '0' == 9) flag = 1;
			pre = num[i] - '0';
		}
		for (int i = 0; i < 10; i++) ans -= dp[n][i];
		ans -= !flag;
		printf("%I64d\n", ans);
	}
	return 0;
}