【数位DP】HDU 2089 不要62

HDU 2089 不要62

题意：

• 求[n, m]中没有连续62并且没有4的数的个数。
• 其中n和m的范围是1e6

解：

• 其实最关键的部分就是dfs的部分。
• 首先我们将每个数位的值保存在一个数组里。
• 解释一下dfs参数的意义：
• pos：第几个数位。例如100，pos分别为2,1,0
• pre：高一位数位的值。例如67，数位0的高一位数位是1，值为6。（它存在的意义是因为62连续是不合法的）
• sta：当前数位的值是否为6（因为当前数位是不是6对下一个低数位的影响是不同的）
• limit：当前数位是否有限制。例如67，对于数位1，如果数位1为6，那么数位0的上限是7。如果数位1是5，那么数位0的上限是9.
• dp[ i ][ j ]：i 指的是第 i 个数位，j指的是数位对应的值是不是6，dp值是该状态下的合法数的数量

注：

• 关于memset优化：因为某个数是不是有62连续，是不是有4，是它自身的性质，不会因为输入的范围而变化，所以放在多组输入的外面。

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <limits>
#include <set>
#include <queue>
#include <vector>
#include <stack>
#include <map>
#define INF 0x3f3f3f3f

using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const ll maxN = 20 + 5;

int n, m;
int dp[maxN][2];
int a[maxN];

int dfs(int pos, int pre, int sta, bool limit)
{
    if(pos == -1)
        return 1;
    if(!limit && dp[pos][sta] != -1)
        return dp[pos][sta];
    int up = limit ? a[pos] : 9;
    int ans = 0;
    for(int i = 0; i <= up; i ++ )
    {
        if(i == 4) continue;
        if(pre == 6 && i == 2) continue;
        ans += dfs(pos - 1, i, i == 6, limit && i == a[pos]);
    }
    if(!limit)
        dp[pos][sta] = ans;
    return ans;
}

int solve(int x)
{
    int pos = 0;
    while(x)
    {
        a[pos ++ ] = x % 10;
        x /= 10;
    }
    return dfs(pos - 1, -1, 0, true);
}

int main()
{
    memset(dp, -1, sizeof(dp));
    while(~scanf("%d%d", &n, &m) && (n || m))
    {
        printf("%d\n", solve(m) - solve(n - 1));
    }
    return 0;
}

数位DP参考博客，写的挺详细的哈哈哈，ORZ

||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||

今天重新做了一遍，发现之前的代码pre和sta其实是重复了，没必要两个参数，完全可以用sta用来记录当前pos的前一个数位是不是6，如果是并且当前位为2那就不合法.qaq
下面是新的代码：
另外附上学长给整理的模板（除了solve里面pos应该初始化为1【他的pos是从1开始的，而我是从0开始的，其实是无所谓啦】，其他都是对的）ps: 就是看了他的模板才想到其实可以sta记录上一位的状态的www我太菜了

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <limits>
#include <set>
#include <queue>
#include <vector>
#include <stack>
#include <map>
#define INF 0x3f3f3f3f

using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int maxN = 1e4 + 5;
int n, m;
int a[35];
int dp[35][2];

int dfs(int pos, int sta, bool limit)//sta:上一位是不是6
{
    if(pos == -1)
        return 1;
    if(!limit && dp[pos][sta] != -1)
        return dp[pos][sta];
    int up = limit ? a[pos] : 9;
    int ans = 0;
    for(int i = 0; i <= up; i ++ )
    {
        if(sta == 1 && i == 2)
            continue;
        if(i == 4)
            continue;
        ans += dfs(pos - 1,i == 6, limit && i == a[pos]);
    }
    if(!limit)
        dp[pos][sta] = ans;
    return ans;
}

int solve(int x)
{
    int pos = 0;
    while(x)
    {
        a[pos ++ ] = x % 10;
        x /= 10;
    }
    return dfs(pos - 1, 0, 1);
}
int main()
{
    memset(dp, -1, sizeof(dp));
    while(~scanf("%d%d", &n, &m) && (n || m))
    {
        printf("%d\n", solve(m) - solve(n - 1));
    }
    return 0;
}