在 Python 中查找具有相同连续差异的数字的程序
假设我们必须找到一个大小为N的数组,使得每两个连续数字之间的绝对差为K。答案中的每个数字都不能有前导零,除了数字0本身。
所以,如果输入像N=4K=7,那么输出将是[1818,2929,7070,8181,9292],这里0707是无效的,因为它有前导0。
示例
让我们看看以下实现以获得更好的理解-
from collections import deque def solve(N, K): if N == 1: return list(range(10)) queue = deque(list(range(1, 10))) for n in range(N - 1): len_queue = len(queue) for j in range(len_queue): num = queue.popleft() lsd = num % 10 if lsd - K >= 0: queue.append( num * 10 + lsd - K ) if K and lsd + K <= 9: queue.append( num * 10 + lsd + K ) return list(queue) N = 4 K = 7 print(solve(N, K))
输入
4, 7输出结果
[1818, 2929, 7070, 8181, 9292]