HDU 3555: Bomb (数位DP)

类型：

数位DP

题意：求1~N内，含有49这个子串的数的个数。N<2^63-1

思路：基础数位DP。第一次做。

dp[i][d] 表示d开头的i位数中，不含49子串的个数

dp[i][d] = ∑dp[i-1][k] (0<=k<=9 && !(d==4 && k==9))

之后求0~N之间不包含49这个子串个数的做法是：

从最高位开始比较，ans += ∑dp[本位][0 ~ (N在本位的数 - 1)]

然后判断，如果本位如果取这个数，是否破坏规则（即包含49），如果是，跳出。

最后，如果N这个数本身不破坏规则，ans++。（算上本身）

最后的答案就是（N+1）-ans // 加1是因为爱，，不，是因为0.

坑点：杭电 I64d .. 做题少啊~~~

代码：

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

long long dp[100][10];

int num[30];

void init() {
for (int i = 0; i < 10; i++) {
dp[1][i] = 1;
}
for (int i = 2; i <= 50; i++) {
for (int d = 0; d < 10; d++) {
dp[i][d] = 0;
for (int k = 0; k < 10; k++) {
if (d == 4 && k == 9) continue;
dp[i][d] += dp[i-1][k];
}
}
}
}

int main() {
int t;
scanf("%d", &t);

init();

while (t--) {
long long N;
cin>>N;

long long ans = 0;
long long tmp = N;
int len = 0;
while (tmp) {
num[++len] = tmp%10;
tmp/=10;
}

bool ok = true;
for (int j = 0; j <= num[len]-1; j++) {
ans += dp[len][j];
}
for (int i = len-1; i >= 1; i--) {
for (int j = 0; j <= num[i]-1; j++) {
ans += dp[i][j];
}
if (num[i+1] == 4 && num[i] == 9) {
ok = false;
break;
}
}
if (ok) ans++;
cout<<N+1-ans<<endl;
}
return 0;
}