You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

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

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.08.28 07:51:26