python杨辉三角_杨辉三角I II

杨辉三角I

题目描述

给定一个非负整数 numRows，生成杨辉三角的前 numRows 行。

在杨辉三角中，每个数是它左上方和右上方的数的和。

示例:

输入: 5
输出:
[
     [1],
    [1,1],
   [1,2,1],
  [1,3,3,1],
 [1,4,6,4,1]
]

---

分析与解答

可以一行一行错位加，当然这里提供更简便的方法。

任取一行描述：[1, 2, 1] 如何得到 [1, 3, 3, 1] ？这样，[0] + [1, 2, 1] 与 [1, 2, 1] + [0] 对应位相加即可。+指的是连接哦~

class Solution:
    def generate(self, numRows: int) -> List[List[int]]:
        if numRows == 0:
            return []
        
        res = [[1]]
        for i in range(1, numRows):
            temp1 = res[-1] + [0]
            temp2 = [0] + res[-1]
            res.append([temp1[i] + temp2[i] for i in range(len(temp1))])
            # # 上面三行，Python大神会这样写
            # res.append([x + y for x, y in zip([0] + res[-1], res[-1] + [0])])
        return res

---

杨辉三角II

题目描述

给定一个非负索引 k，其中 k ≤ 33，返回杨辉三角的第 k 行。

在杨辉三角中，每个数是它左上方和右上方的数的和。

示例:

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

进阶：

你可以优化你的算法到 O(k) 空间复杂度吗？

---

分析与解答

不解释。

class Solution:
    def getRow(self, rowIndex: int) -> List[int]:
        cur = [1]
        for _ in range(rowIndex):
            cur1 = [0] + cur
            cur2 = cur + [0]
            cur = [x + y for x, y in zip(cur1, cur2)]

        return cur