LeetCode：119（Python）—— 杨辉三角 II（简单）

娱乐不打烊丶

已于 2022-12-08 11:05:03 修改

阅读量5.7w

点赞数 1

文章标签： leetcode 算法职场和发展数据结构 python

于 2022-10-22 18:32:56 首次发布

本文链接：https://blog.csdn.net/m0_61661179/article/details/127465340

版权

杨辉三角 II

概述：给定一个非负索引 rowIndex，返回「杨辉三角」的第 rowIndex 行。在「杨辉三角」中，每个数是它左上方和右上方的数的和。

输入: rowIndex = 3
输出: [1,3,3,1]

输入: rowIndex = 0
输出: [1]

输入: rowIndex = 1
输出: [1,1]

方法一：数学法

思路：每行数字左右对称，由 1 开始逐渐变大再变小，并最终回到 1。每个数字等于上一行的左右两个数字之和，可用此性质写出整个杨辉三角。

# 数学法
class Solution:
    def getRow(self, rowIndex: int) -> List[int]:
        ans = []
        n = rowIndex
        for i in range(n + 1):
            row = []
            for j in range(0, i + 1):
                if j == 0 or j == i:
                    row.append(1)
                else:
                    row.append(ans[i - 1][j] + ans[i - 1][j - 1])
            ans.append(row)
        return ans[-1]

方法二：错位相加

思路：观察一下规律，发现当前一行只比上一行多了一个元素，最最关键的一点：本行元素等于上一行元素往后错一位再逐个相加。因此我们只要对最后一行单独处理：最后一行首、尾分别添加一个零然后对应位置求和就可以得到新的一行。

# 错位相加
class Solution:
    def getRow(self, rowIndex: int) -> List[int]:
        n = rowIndex
        if n == 0:
            return [1]
        ans = [[1], ]
        while len(ans) < (n + 1):
            row = [i + j for i, j in zip([0] + ans[-1], ans[-1] + [0])]
            ans.append(row)
        return ans[-1]