Python如何生成由1~n不重复数字组成的所有n位数字
实现思路
你要生成的这类数字,本质上就是1到n这n个正整数的全排列拼接结果——因为要求每个数字恰好出现一次、每一位都从1~n里选,刚好符合全排列的定义。n最大为9的时候总共有9! = 362880个结果,不管用哪种方法生成性能都不会有问题。
基础实现代码
方法1:调用标准库(最简便)
Python标准库itertools里的permutations天生就是用来生成不重复元素排列的,几行代码就能搞定:
from itertools import permutations def generate_n_digit_numbers(n: int) -> list[int]: # 参数合法性校验 if not isinstance(n, int) or n < 1 or n >= 10: raise ValueError("n必须取值为1~9的整数") res = [] # 生成1~n所有元素的全排列 for perm in permutations(range(1, n+1)): # 把排列元组拼接为整数,比如(2,3,1)转成231 num = int(''.join(str(d) for d in perm)) res.append(num) return res # 示例:n=3时输出 [123, 132, 213, 231, 312, 321] print(generate_n_digit_numbers(3))
这个方法的底层是C实现的,运行效率很高,日常用优先选这个就行。
方法2:手写回溯(方便自定义规则)
如果你想自己实现逻辑、或者后续要加复杂的自定义约束,可以用回溯法,核心思路是逐位选数字,用过的数字标记为已使用,选满n位就存结果,再回退状态选其他可能:
def generate_n_digit_numbers_backtrack(n: int) -> list[int]: if not isinstance(n, int) or n < 1 or n >= 10: raise ValueError("n必须取值为1~9的整数") res = [] used = [False] * (n + 1) # 下标对应数字,标记是否已经被选用 def dfs(current: int): # 已经凑够n位,存入结果 if len(str(current)) == n: res.append(current) return # 尝试选每一个没被用过的数字 for d in range(1, n+1): if not used[d]: used[d] = True dfs(current * 10 + d) used[d] = False # 回溯,撤销选择 dfs(0) return res
拓展到其他类似生成需求
这套思路可以覆盖绝大多数“按规则生成指定位数数字”的需求,只需要对应调整规则即可,举几个常见例子:
- 可选数字不是1~n:把代码里遍历的
range(1, n+1)换成你指定的数字集合即可,如果集合里包含0,额外加个判断:第一位不能选0,否则生成的会是不足n位的数字。 - 允许数字重复使用:标准库方案把
permutations换成itertools.product,指定repeat=n即可;回溯方案直接去掉used数组的标记逻辑,每次递归都可以选所有允许的数字。 - 加额外约束(比如相邻位差值不超过2、数字为偶数、各位和为固定值等):如果用标准库方案,在把排列转成数字之后加规则判断,符合要求再加入结果;如果用回溯方案,可以在递归选数字的时候就提前剪枝——比如要求是偶数,最后一位只选偶数,直接跳过不符合要求的分支,运行效率会高很多。
内容的提问来源于stack exchange,提问作者amirhosseinshj
相关产品推荐
相关产品推荐

