leetcode_967 连续差相同的数字

271 阅读1分钟

要求

返回所有长度为 n 且满足其每两个连续位上的数字之间的差的绝对值为 k 的 非负整数 。

请注意,除了 数字 0 本身之外,答案中的每个数字都 不能 有前导零。例如,01 有一个前导零,所以是无效的;但 0 是有效的。

你可以按 任何顺序 返回答案。

示例 1:

输入:n = 3, k = 7
输出:[181,292,707,818,929]
解释:注意,070 不是一个有效的数字,因为它有前导零。

示例 2:

输入:n = 2, k = 1
输出:[10,12,21,23,32,34,43,45,54,56,65,67,76,78,87,89,98]

示例 3:

输入:n = 2, k = 0
输出:[11,22,33,44,55,66,77,88,99]

示例 4:

输入:n = 2, k = 2
输出:[13,20,24,31,35,42,46,53,57,64,68,75,79,86,97]

提示:

  • 2 <= n <= 9
  • 0 <= k <= 9

核心代码

class Solution:
    def numsSameConsecDiff(self, n: int, k: int) -> List[int]:
        if n == 1:
            return list(range(10))
        res = list(map(str,range(1,10)))
        for i in range(n-1):
            t = []
            for x in res:
                a = int(x[-1])
                if a - k >= 0:
                    t.append(x + str(a - k))
                if a + k <= 9 and k != 0:
                    t.append(x + str(a + k))
            res = t
        return [int(i) for i in res]

image.png

解题思路:如果N=1,返回0到9,一位的时候就没有前后差的事,所以直接返回就行了;如果N>1, 第一位数可以是1到9,一位一位增加,下一位可以是当前位-K或者当前位+K,只要在0到9之间即可。