0Pricing
Coding Interview Prep · 课时

使用 itertools 枚举

笛卡尔积、组合与排列

使用 itertools 枚举 是 CoddyKit 上的免费 Coding Interview Prep 课时。 这是第 2 节课,共 4 节。 你可以在下方免费阅读本课时的完整内容 — 然后在浏览器中使用内置代码编辑器和全天候 AI 导师进行实践。 这是 Coding Interview Prep 学习路径的一部分,你的进度在网页和 CoddyKit 应用中同步。 Coding Interview Prep 课程共包含 4 节课。

让迭代工具库处理循环

Python 的迭代工具库会为您生成组合和排列,因此您可以少写循环代码,也能减少错误。🙂

先导入迭代工具库

所有功能都位于同一个模块中,因此请先导入迭代工具库。然后您就可以直接调用 product、permutations 和 combinations。

import itertools

用 product 替代嵌套循环

使用 product 可以替代针对相同选项的多层嵌套循环。它会生成给定范围内所有选取结果组成的元组。

for combo in itertools.product(range(3), repeat=2):
    print(combo)

用 permutations 生成排列

当顺序很重要时,permutations会列出元素的每种排列。它非常适合处理小规模、类似旅行商问题的暴力搜索。

for p in itertools.permutations([1, 2, 3]):
    print(p)

用 combinations 进行选择

当顺序不重要时,combinations会选出每个大小为 r 的组合。它非常适合选择一组元素进行测试。

for c in itertools.combinations([1, 2, 3, 4], 2):
    print(c)

排列与组合的区别

关键区别在于顺序:permutations 会将 (1,2) 和 (2,1) 视为不同结果,而 combinations 会将它们视为同一种选择。

注意增长速度

排列数量按 n! 增长,速度会迅速膨胀。即使 n 等于 11,也已经超过四千万种排列,因此请让 n 保持较小。

它们采用惰性生成

这些函数是生成器:它们一次生成一个元素,而不是构建一个巨大的列表,因此循环时占用的内存很少。

结合检查条件

先生成候选项,然后只保留通过测试的候选项。这种模式可以将困难的搜索变成简短易读的循环。

best = min(itertools.permutations(cities), key=tour_length)

允许重复的组合

需要选择可以重复的元素吗?请使用允许重复的组合功能,让同一个元素可以在一个组合中出现多次。

选择正确的工具

请问自己两个问题:顺序重要吗?元素可以重复吗?答案会直接指向正确的迭代工具库函数。

快速检查

您必须枚举从 6 个元素中选择 3 个元素的所有方式,并且顺序不重要。

回顾

使用 product、permutations 和 combinations 可以清晰地完成枚举。请根据顺序是否重要以及元素是否重复来进行选择,同时注意数量增长。🚀

常见问题解答

「使用 itertools 枚举」课时是免费的吗?

是的 — 「使用 itertools 枚举」的完整文本可在网页上免费阅读。要进行交互式练习(内置代码编辑器和全天候 AI 导师)并解锁 Coding Interview Prep 课程的其余内容,请升级到 CoddyKit PRO。 Coding Interview Prep 课程共包含 4 节课。

「使用 itertools 枚举」这节课中我会学到什么?

笛卡尔积、组合与排列 你通过在浏览器中直接运行的动手代码来练习 Coding Interview Prep,全天候 AI 导师会在你学习这节课的过程中回答你的问题。

学习 Coding Interview Prep 需要有经验吗?

无需任何先前经验。CoddyKit 上的 Coding Interview Prep 课程适合初学者到高级学习者,你可以从这里开始或从头开始,按照自己的节奏学习。 这是第 2 节课,共 4 节。

「使用 itertools 枚举」课时需要多长时间?

大多数 CoddyKit 课程大约需要 5–10 分钟。每节课都很精短且互动,所以你能稳步进步,并在网页和应用中从离开的地方继续。

我能在这节 Coding Interview Prep 课中编写并运行代码吗?

能。每节 Coding Interview Prep 课都包含内置代码编辑器,你可以在浏览器中直接编写并运行真实代码,并获得即时 AI 反馈 — 无需本地设置。

此课程中的所有课时

  1. 暴力搜索也是有效策略
  2. 使用 itertools 枚举
  3. 使用位掩码枚举子集
  4. 巧妙缩小搜索空间
← 返回 Coding Interview Prep