python 刷Leetcode: Pascal’s Triangle II
题目描述
Given an index k, return the kth row of the Pascal’s triangle.
For example, given k = 3,
Return [1,3,3,1].
Note:
Could you optimize your algorithm to use only O(k) extra space?
即返回第k行的杨辉三角
杨辉三角(Pascal’s Triangle)描述
1
1 1
1 2 1
1 3 3 1
1 4 6 4 1
1 5 10 10 5 1
为更好的发现规律,改写:
1 0
1 1 0
1 2 1 0
1 3 3 1 0
1 4 6 4 1 0
1 5 10 10 5 1 0
0是为了构造而添加的辅助的,这样下一行的值就是对应上一行的值加它前面的值.
例如:第k行第i行L[k][i]=L[k-1][i]+L[k-1][i-1]
首先需要实现杨辉三角的函数,为节省内存,可以利用yield 生成器。
def pascal(self):
L = [1]
while True:
yield L
L.append(0)
L=[L[i]+L[i-1] for i in xrange(len(L))]
正式代码:
class Solution(object):
def getRow(self, rowIndex):
"""
:type rowIndex: int
:rtype: List[int]
"""
ans =[]
f1 = self.pascal()
for x in xrange(rowIndex+1):
ans = f1.next()
return ans
def pascal(self):
L = [1]
while True:
yield L
L.append(0)
L = [L[i - 1] + L[i] for i in xrange(len(L))]
备注:
刚开始我用了:
for x in xrange(rowIndex+1):
ans = self.pascal().next()
return ans
这就错了,因为这里for循环每次调用self.pascal().next()都是从初始开始,即得到的结果总是L=[1],解决的方法是先定义一个引用f1,然后每次循环调用的都是f1引用,这样才能next方法才能往下生成我们想要的结果。