HDU 2089 不要62(数位dp&记忆化搜索)

Posted 2020-06-13 shiyicode

题目链接：[kuangbin带你飞]专题十五 数位DP C - 不要62

题意

杭州人称那些傻乎乎粘嗒嗒的人为62（音：laoer）。
杭州交通管理局经常会扩充一些的士车牌照，新近出来一个好消息，以后上牌照，不再含有不吉利的数字了，这样一来，就可以消除个别的士司机和乘客的心理障碍，更安全地服务大众。
不吉利的数字为所有含有4或62的号码。例如：
62315 73418 88914
都属于不吉利号码。但是，61152虽然含有6和2，但不是62连号，所以不属于不吉利数字之列。
你的任务是，对于每次给出的一个牌照区间号，推断出交管局今次又要实际上给多少辆新的士车上牌照了。

思路

dp[i][j]表示当前为第i位且上一位(是／不是)6的数的个数
知道上一位是否为六，对当前为进行if(i!=4 && !(pre&&i==2))即可筛选。记忆化搜索即可。

代码

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <vector>

using namespace std;

int dp[10][2];
int dis[10];

int dfs(int len, bool pre, bool flag)
{
    if(len < 0)
        return 1;
    if(!flag && dp[len][pre]!=-1)
        return dp[len][pre];
    int ans = 0;
    int end = flag?dis[len]:9;
    for(int i=0; i<=end; i++)
    {
        if(i!=4 && !(pre&&i==2))
            ans += dfs(len-1, i==6, flag&&i==end);
    }
    if(!flag)
        dp[len][pre] = ans;
    return ans;
}

int solve(int n)
{
    int len = 0;
    while(n)
    {
        dis[len++] = n%10;
        n /= 10;
    }
    return dfs(len-1, 0, 1);
}

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