杭电 hdu 2089 不要62【数位dp】

题目链接:https://cn.vjudge.net/contest/247095#problem/B

题目大意:给出区间【n,m】,问不包含数字4和62的个数

思路:

 dp[i][0],表示长度为i,不存在不吉利数字
 dp[i][1],表示长度为i,不存在不吉利数字,且最高位为2
 dp[i][2],表示长度为i,存在不吉利数字

dp[i][0]=dp[i-1][0]*9-dp[i-1][1];//在最高位加上除4以外的9个数字,但要减掉2之前加上6

 dp[i][1]=dp[i-1][0];//在不含不吉利数字的最高位加上2
 dp[i][2]=dp[i-1][2]*10+dp[i-1][0]+dp[i-1][1];
  //在已有不吉利数字前加任意数字,或者无不吉利数字的最高位加4,或者在2前面加6

在solve函数中分为四者情况判断:

高位已经出现过62或4,当前位出现4,当前位加上一位组成62,当前位加下一位组成62.

#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <string.h>
using namespace std;
int dp[10][3];
/*
 * dp[i][0],表示长度为i,不存在不吉利数字
 * dp[i][1],表示长度为i,不存在不吉利数字,且最高位为2
 * dp[i][2],表示长度为i,存在不吉利数字
 */
void init()
{
    dp[0][0]=1;dp[0][1]=0;dp[0][2]=0;
    for(int i=1;i<=6;i++)
    {
        dp[i][0]=dp[i-1][0]*9-dp[i-1][1];//在最高位加上除4以外的9个数字,但要减掉2之前加上6
        dp[i][1]=dp[i-1][0];//在不含不吉利数字的最高位加上2
        dp[i][2]=dp[i-1][2]*10+dp[i-1][0]+dp[i-1][1];
        //在已有不吉利数字前加任意数字,或者无不吉利数字的最高位加4,或者在2前面加6
    }
}
int bit[10];
int solve(int n)
{
    int len=0;
    int tmp=n;
    while(n)
    {
        bit[++len]=n%10;
        n/=10;
    }
    bit[len+1]=0;
    int ans=0;
    bool flag=false;
    for(int i=len;i>=1;i--)
    {
        ans+=dp[i-1][2]*bit[i];
        if(flag)//高位已经出现4或者62,后面随意
            ans+=dp[i-1][0]*bit[i];
        if(!flag&&bit[i]>4)
            ans+=dp[i-1][0];
        if(!flag&&bit[i+1]==6&&bit[i]>2)
            ans+=dp[i][1];
        if(!flag&&bit[i]>6)
            ans+=dp[i-1][1];
        if(bit[i]==4||(bit[i+1]==6&&bit[i]==2))
            flag=true;
    }
    if(flag)ans++;//这个数本身
    return tmp-ans;
}
int main()
{
    init();
    int n,m;
    while(scanf("%d%d",&n,&m)==2)
    {
        if(n==0 && m==0)break;
        printf("%d\n",solve(m)-solve(n-1));
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_36782366/article/details/81706940