HDU2089(数位DP)

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

第一种方式：递推，累加。

区间 [n, m] 中满足条件的个数等于[0, m]的个数减去[0, n-1]的个数。

怎么计算区间[0,n]中满足条件的个数呢？

预处理出数组 f[i][j]，表示长度为 i 并且以 j 开头（可以有前导0）的不含数字 4 和 62 的数字的个数，其实可以说是区间[0, 10^i-1]中不含 4 和 62 的数字的个数。

从高位开始枚举 n 的每一位（比如第 i 位），然后枚举比该位数字小的数作（用 j 枚举）为开头，累加相应长度的个数（即累加 f[i][j]，前提是在第 i 位的左边没有出现4 和 62）。因为这里枚举的每一位的数字都要比该位的数字要小，所以累加出来的结果不包括数字n本身（如果这个数字合法的话），所以这个统计的区间是 [0, n) 的，所以最终的结果是[0, m)-[0,m)。

那怎么处理 f[i][j] 呢？见代码！

#include<iostream>
#include<sstream>
using namespace std;

int f[8][10];

void init()
{
    f[0][0] = 1;
    for (int i=1; i<=7; i++)
        for (int j=0; j<10; j++)  //枚举第 i 位的数字
            for (int k=0; k<10; k++)  //枚举第 j 位的数字
                if (j != 4 && !(j == 6 && k == 2))
                    f[i][j] += f[i-1][k];

}

int calc(int n)
{
    stringstream in;
    in<<n;
    string str;
    in>>str;

    int digit[10] = {0};

    int len = str.size();
    for (int i=0; i<str.size(); i++)
        digit[len-i] = str[i] - '0';

    int ans = 0;
    for (int i=len; i>=1; i--)
    {
        for (int j=0; j<digit[i]; j++)
            if (j != 4 && !(j==2 && digit[i+1] == 6))
                ans += f[i][j];
        if (digit[i] == 4 || (digit[i] == 2 && digit[i+1] == 6))
            break;
    }

    return ans;
}

int main()
{
    int n, m;

    init();

    while (cin>>n>>m && n && m)
    {
        cout<<calc(m+1)-calc(n)<<endl;  //[0, m)-[0,m)
    }

    return 0;
}

第二种方式：记忆化搜索。

设函数 int dfs(int pos, int pre, bool limit)表示处理剩余的pos位（从低位到高位分别为第 1 到 len 位），上一位数字位 pre，对当前位的处理 有/没有 限制的方案数，也可以说是上一位为pre并且 有/没有 限制时，计算长度为pos的合法的方案数。

在函数内部枚举当前位所有可能的数字，用类似于第一种方式中的累加的方法来累加方案。

#include<iostream>
using namespace std;

int dp[15][10];
int digit[15];
int len;

int dfs(int pos, int pre, bool limit)  //上一位为pre并且 有/没有(limit = true/false) 限制时，计算长度为pos的合法的方案数
{
    if (pos == 0)
        return 1;
    if (!limit && dp[pos][pre])   //记忆化
        return dp[pos][pre];

    int ans = 0;
    int Max = limit ? digit[pos] : 9;  //确定当前位的最大范围
    for (int i=0; i<=Max; i++)
    {
        if (i == 4) continue;     //不能含有"4"
        if (pre == 6 && i == 2) continue;  //不能含有"62"
        ans += dfs(pos-1,i,limit&&(i==Max));  //累加
    }

    if (!limit)    //dp[][]储存的是没有限制的方案数
        dp[pos][pre] = ans;
    return ans;
}

int calc(int n)
{
    len = 0;

    while (n > 0)
    {
        digit[++len] = n % 10;
        n /= 10;
    }

    return dfs(len,0,true);
}

int main()
{
    int n, m;

    while (cin>>n>>m && n && m)
    {
        cout<<calc(m)-calc(n-1)<<endl;  //在第一种方式中，在统计方案的时候枚举每一位的数字的时候上限是不能等于当前位的数字的，所以统计的时候不包含 n 本身，但此时的dfs中对于每一位数字的枚举是可以等于当前数字的，所以统计的时候包含 n 本身，这方面的不同决定了统计时的区间的开闭情况，也就决定了到底是calc(m+1)-calc(n)还是calc(m)-calc(n-1)。

    }

    return 0;
}