刷题1

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方法才能往下生成我们想要的结果。