生成器版本的杨辉三角

📌 下边代码为生成器版本的杨辉三角

python 复制代码
def triangles():
    row = [1]
    while True:
        yield row
        row = [1] + [row[i - 1] + row[i] for i in range(1, len(row))] + [1]

n = 0
results = []
for t in triangles():
    results.append(t)
    n += 1
    if n == 10:
        break

for t in results:
    print(t)

🔍 核心函数解释:triangles()

第一行初始化:

python 复制代码
row = [1]

设置杨辉三角的第一行。

无限生成:

python 复制代码
while True:
    yield row

使用生成器 yield,每次返回当前行 row

更新下一行:

python 复制代码
row = [1] + [row[i - 1] + row[i] for i in range(1, len(row))] + [1]

该行生成下一行杨辉三角,具体包括:

  • 开头加 1
  • 中间是相邻元素相加形成的新列表
  • 结尾加 1

✨ 重点解析:中间部分 [row[i - 1] + row[i] for i in range(1, len(row))]

这是杨辉三角中间项的生成规则。

  • 遍历 i 从 1 到 len(row) - 1
  • 每次将相邻两个数相加:row[i - 1] + row[i]

示例:

假设 row = [1, 3, 3, 1]

python 复制代码
中间部分计算如下:
i = 1: 1+3 = 4
i = 2: 3+3 = 6
i = 3: 3+1 = 4

结果为:[4, 6, 4]

下一行变为:

python 复制代码
[1] + [4, 6, 4] + [1] → [1, 4, 6, 4, 1]

🤔 为什么第二次输出是 [1, 1]

第一次生成的是:

python 复制代码
row = [1]

进入更新逻辑:

python 复制代码
[row[i - 1] + row[i] for i in range(1, len(row))]

此时 len(row) = 1,所以:

python 复制代码
range(1, 1) → 空序列 → [](中间部分为空)

最终:

python 复制代码
row = [1] + [] + [1] = [1, 1]

因此第二次打印是:

python 复制代码
[1, 1]

✅ 结果回顾(前 10 行)

csharp 复制代码
[1]
[1, 1]
[1, 2, 1]
[1, 3, 3, 1]
[1, 4, 6, 4, 1]
[1, 5, 10, 10, 5, 1]
[1, 6, 15, 20, 15, 6, 1]
[1, 7, 21, 35, 35, 21, 7, 1]
[1, 8, 28, 56, 70, 56, 28, 8, 1]
[1, 9, 36, 84, 126, 126, 84, 36, 9, 1]

📚 小结

  • range(1, len(row)) 是为了获取所有 相邻元素组合
  • row 只有一个元素时,中间列表为空,因此生成 [1, 1]
  • 该生成器用 yield 无限生成杨辉三角的每一行,直到外部主动 break
相关推荐
CM莫问3 小时前
<论文>(微软)WINA:用于加速大语言模型推理的权重感知神经元激活
人工智能·算法·语言模型·自然语言处理·大模型·推理加速
计信金边罗5 小时前
是否存在路径(FIFOBB算法)
算法·蓝桥杯·图论
MZWeiei5 小时前
KMP 算法中 next 数组的构建函数 get_next
算法·kmp
Fanxt_Ja6 小时前
【JVM】三色标记法原理
java·开发语言·jvm·算法
luofeiju6 小时前
行列式的性质
线性代数·算法·矩阵
緈福的街口6 小时前
【leetcode】347. 前k个高频元素
算法·leetcode·职场和发展
pen-ai7 小时前
【统计方法】基础分类器: logistic, knn, svm, lda
算法·机器学习·支持向量机
鑫鑫向栄7 小时前
[蓝桥杯]春晚魔术【算法赛】
算法·职场和发展·蓝桥杯
roman_日积跬步-终至千里7 小时前
【Go语言基础【3】】变量、常量、值类型与引用类型
开发语言·算法·golang
FrankHuang8888 小时前
使用高斯朴素贝叶斯算法对鸢尾花数据集进行分类
算法·机器学习·ai·分类