338. 计数问题

在这里插入图片描述
在这里插入图片描述
思路:
个人认为最难理解的应该是dp[N][N]这个元素的作用,这个元素实际上的作用其实是防止dfs的时候重复计算很多相同的步骤,当dp != -1时说明,这个地方在没有前导零和无限制的情况下已经计算过了,后面的可以直接使用计算过的数字来使用,看过很多帖子都美誉说明dp数组的作用,dp数组的含义是回溯的关键
代码:

# include<iostream>
# include<algorithm>
# include<cstdio>
# include<cstring>
using namespace std;

const int N = 15;
int l,r,a[N],len,dp[N][N];
//dp[N][N]是真正的dp思想所在,表示该状态是否被算过

/*
pos 表示数字的位数
limit 和上界数字是否相等
lead 前导零是否存在,lead = 1存在前导零,否则不存在
sum 用到次数总和
num 表示哪个数字
*/
int dfs(int pos,int sum,int num,int lead,int limit)//limit = 1代表和上界数字相等
{
    //边界条件
    if(!pos)
    {
        if(lead && !num) return 1;
        return sum;
    }
    //只有无限制、无前导零才算,不然都是未搜索完的情况
    //dp[pos][sum] != -1代表算过
    if(!limit && !lead && dp[pos][sum] != -1) return dp[pos][sum];
    //res 来记录答案 up表示能枚举的最高为数
    int res = 0,up = limit ? a[pos] : 9;
    for(int i = 0;i <= up;i++)
    {
        int t;
        if(i == num)
        {
            //如果是0
            if(num == 0)
            {
                t = sum + (lead == 0);//存在前导零的话,不算进去
            }
            else
            {
                t = sum + 1;
            }
        }
        else
        {
            t = sum;
        }
        //limit && i == up代表当前是否是约束条件而且选择的i是否为最大的数
        res += dfs(pos - 1,t,num,lead && i == 0,limit && i == up);
    }
    return limit ? res : (lead ? res : dp[pos][sum] = res);
}

int cal(int x,int num)
{
    memset(dp,-1,sizeof dp);
    len = 0;
    while(x) a[++len] = x % 10, x /= 10;
    return dfs(len,0,num,1,1);
}

int main()
{
    while(cin >> l >> r,l || r)
    {
        if(l > r) swap(l,r);
        for(int i = 0;i <= 9;i++)
        {
            cout << cal(r,i) - cal(l - 1,i) << " ";
        }
        cout << endl;
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值