119. 楊輝三角 II

給定一個非負索引 k，其中 k ≤ 33，返回楊輝三角的第 k 行。

在楊輝三角中，每一個數是它左上方和右上方的數的和。

 

 

 1 class Solution(object):
 2     def getRow(self, rowIndex):
 3         """
 4         :type rowIndex: int
 5         :rtype: List[int]
 6         """
 7         a = []
 8         # 初始化
 9         for i in range(1, rowIndex + 2):
10             a.append([0] * i)
11         # 最頂端元素
12         a[0][0] = 1
13         for i in range(1, rowIndex+1):
14             for j in range(i + 1):
15                 # 每一行第一個和最後一個元素都是1
16                 if j == i or j == 0:
17                     a[i][j] = 1
18                 # 不然是肩上兩個元素之和
19                 else:
20                     a[i][j] = a[i - 1][j] + a[i - 1][j - 1]
21         # 返回最後一個list
22         return a[-1]
23 
24 
25 if __name__ == '__main__':
26     solution = Solution()
27     print(solution.getRow(3))