1. 程式人生 > >leetcode119. 楊輝三角 II

leetcode119. 楊輝三角 II

https://leetcode-cn.com/problems/pascals-triangle-ii/

給定一個非負索引 k,其中 k ≤ 33,返回楊輝三角的第 k 行。
在楊輝三角中,每個數是它左上方和右上方的數的和。
示例:
輸入: 3
輸出: [1,3,3,1]
進階:
你可以優化你的演算法到 O(k) 空間複雜度嗎?

注意k==0代表第一行,找個tmp儲存上一級的結果就行:

class Solution:
    def getRow(self, rowIndex):
        """
        :type rowIndex: int
        :rtype: List[int]
        """
        for i in range(rowIndex+1):
            row = [1] * (i+1)
            if i >= 2:
                for j in range(1,i):
                    row[j] = tmp[j-1] + tmp[j]
            tmp = row
        return tmp

二項式定理:

class Solution:
    def getRow(self, rowIndex):
        """
        :type rowIndex: int
        :rtype: List[int]
        """
        row = [1] * (rowIndex+1)
        if rowIndex >= 1:
            for i in range(1, rowIndex+1):
                row[i] = row[i-1]*(rowIndex+1-i)//i
        return row