数字游戏(数位dp)

1. 问题描述:

科协里最近很流行数字游戏。某人命名了一种不降数,这种数字必须满足从左到右各位数字成大于等于的关系,如123,446。现在大家决定玩一个游戏,指定一个整数闭区间[a,b],问这个区间内有多少个不降数。
输入格式:
题目有多组测试数据。每组只含2个数字a, b (1 <= a, b <= 2^31)。
输出格式
每行给出一个测试数据的答案,即[a, b]之间有多少不降数。
样例输入
1 9 
1 19
样例输出
9
18

来源:https://www.acwing.com/activity/content/11/

2. 思路分析:

① 这道题目属于数位dp知识点的题目。对于数位dp的题目处理的思路都是类似的,首先是根据题目的数据范围声明一个数组或者列表f(大部分是二维数组或者列表),使用一个init方法来初始化数组或者列表的值,对于这道题目来说可以声明一个二维列表(python语言),其中f[i][j]表示的意思是当前i位数并且最高位为j的数目,一开始的时候可以只有一位那么最高位为0-9,所以f[1][i] = 1,我们可以从第二位数字开始递推,使用三层循环表示递推的过程,第一层循环表示当前有i位数字,第二层循环表示当前i位数字最高位为j,第三层循环表示i - 1位数字的最高位数字为k,其实举个例子就很好理解了,比如一开始的时候有一位数字那么第一位数字的情况为:0 1 2 3 4 5 6 7 8 9,递推两位数字的时候,如果最高位为0那么有00,01,02,03,,,,09...。第二步是使用一个dp方法来求解区间[0, t]方案数目f(t),区间[x, y]的方案数目为f(y) - f(x)。

② 在dp方法中首先需要使用列表nums存储当前的数字n的各个位上的数字(除10取余求解n中各个位置上的数字),从高位到低位开始遍历n中各个位置上的数字,也即逆序遍历,当前的位置上的数字为x = nums[i],对于当前的这一位我们可以考虑填入0~x - 1中数字,可以使用循环来累加当前最高位为j(j的范围为[0, x - 1])并且总共位置有i + 1位的二维列表f[i + 1][j]的值,这也是在一开始的时候使用init方法进行预处理的原因,在这里我们可以直接取出f中对应位置的值即可,这样循环累加的就是有i + 1位数字并且最高位为j的数目,然后考虑当前位填入数字x的情况,使用一个变量last来记录上一次填入的最大值,这样在进入循环判断nums中下一个数字nums[i - 1]的时候第i位为固定数字x也即考虑第i位为x的情况,这样下一位尝试的数字范围为[last, x),位数为当前的下标i。其实举具体的例子会更好理解一点。

3. 代码如下:

N = 15
# f[i][j]表示一共有i位, 而且最高位填j的数的个数
f = [[0] * N for i in range(N)]


# 预处理得到f列表对应位置的值
def init():
    for i in range(10):
        f[1][i] = 1
    for i in range(2, N):
        for j in range(10):
            for k in range(j, 10):
                f[i][j] += f[i - 1][k]


# 数位dp一般都有last变量, 不同的题目last的含义是不一样的
def dp(n: int):
    if n == 0: return 1
    nums = list()
    while n:
        nums.append(n % 10)
        n //= 10
    res = 0
    last = 0
    for i in range(len(nums) - 1, -1, -1):
        x = nums[i]
        for j in range(last, x):
            res += f[i + 1][j]
        # 下一个尝试的数字比上一次最后一个数字要小直接break
        if x < last: break
        last = x
        # 最右边的分支, 填入的最后一个数字(每一次轮到下一个数字的时候都是固定上一位最大的数字所以到i=0的时候说明需要将最后一个符合条件的数字累加到结果中)
        if i == 0: res += 1
    return res


if __name__ == '__main__':
    init()
    while True:
        l, r = map(int, input().split())
        print(dp(r) - dp(l - 1))

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值