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

时间:2022-12-16 09:17:10

题目链接[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;
}