HDOJ 2089 不要62 (数位DP)

第一道数位DP的题目.（后来发现这题可以不用DP水过。。。）

还是花了挺长时间的，原理不难懂，但是细节总是有问题。

题意：给定一个区间，求该区间没有 "4" 或在一起的 "62" 的数的个数。

dp[i][j] 表示 i 位数，j 开头满足条件的数的个数。

则 dp[i][j] = sigma(k = 0 to 9) (dp[i-1][k])     (j != 4, !(j == 6 && k == 2))

利用前缀和的思想，区间 [n,m] 的答案即为 cnt[(m) - cnt(n-1)

参考了网上的代码。。

用类似“低位对低位，高位对高位”的方式，用 digit[] 数组存数字，然后枚举。

比如一个数是 3xxx 那么 所有的 1xxx, 2xxx 0xxx 满足的就都要加上。

但是如果碰到 62 或者 4 了，比如 36245

在处理4这位的时候，不知道前面有 62 ，所以把 3623xx ... 3620xx 都算上了，所以到这里要把之前的一并减去。

另一个更好的办法是从高位开始算，碰到不合法的直接 break 。

代码：

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

int n, m;
const int maxn = 1000100, maxd = 8;
int dp[maxd][10];

int getn(int num)
{
    int digit[10]={0};
    int len = 0, ret = 0;
    while(num) {
        digit[len++] = num%10;
        num/=10;
    }
    bool ok = 1;
    for(int i = len-1; i >= 0; i--) {
        for(int j = 0; j < digit[i]; j++) {
            if(j != 4 && !(i != len-1 && j == 2 && digit[i+1] == 6))
                ret += dp[i+1][j];
        }
        if((digit[i] == 4) || (i != len-1 && digit[i] == 2 && digit[i+1] == 6)) {
            ok = 0;
            break;
        }
    }
    if(ok && digit[0] != 4 && !(digit[0] == 2 && digit[1] == 6)) ret++;

    return ret;
}

int main()
{
//    freopen("1","w",stdout);
    memset(dp, 0, sizeof(dp));
    for(int i = 0; i <= 9; i++)
        if(i != 4) dp[1][i] = 1;
    for(int i = 2; i <= 7; i++) {//枚举位数
        for(int j = 0; j <= 9; j++) {//枚举最高位
            for(int k = 0; k <= 9; k++) {//枚举第二位
                if(j != 4 && !(j == 6 && k == 2)) {
                    dp[i][j] += dp[i-1][k];
                }
            }
        }
    }
    while(cin >> n >> m && (n || m)) {
//    for(n = 999900, m = 1000000; n <= m; n++) {
        cout << getn(m) - getn(n-1) << endl;
    }

    return 0;
}