题目:Given an index k, return the kth row of the Pascal’s triangle.
For example, given k = 3,
Return [1,3,3,1]
解题思路:由于只需要返回第k行的帕斯克三角,所以用一个数组p储存。
帕斯克三角:
1
1 1
1 2 1
1 3 3 1
1 4 6 4 1
可以看出:a[i][j]=a[i-1][j-1]+a[i-1][j]
所以,对于第k行的第j个元素,a[j]=上一行的(a[j]+a[j-1])
但是要注意,在循环时,要从后向前循环,否则当循环到下一个元素时,上一行的a[j]会被本行的a[j]覆盖,如:
第k行:a=[1,2,1]
第k+1行:
a[1]=a[0]+a[1]=1+2=3
a[2]=a[1]+a[2]=3+1=4 #此时的a[1]已经变为本行计算出的a[1]
代码:
class Solution(object):
def getRow(self, rowIndex):
p=[1]*(rowIndex+1) #初始化数组p
for i in range(2,rowIndex+1):
for j in range(i-1,0,-1): # j属于[i-1,1]
p[j] = p[j]+p[j-1]
return P