119. Pascal's Triangle II [easy] (Python)

题目链接

https://leetcode.com/problems/pascals-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,返回帕斯卡三角形的第k行。比如给定k=3,返回[1,3,3,1]。
备注:你能仅用O(k)的额外空间实现你的算法吗?

思路方法

如果你不太了解帕斯卡三角形,可先参考这一题:
118. Pascal’s Triangle [easy] (Python)

思路一

用到一个小小的trick,即:每一行的结果可以由上一行和上一行的偏移相加得到。例如求第4行:

 1 3 3 1 0 + 0 1 3 3 1 = 1 4 6 4 1 

代码

class Solution(object):
    def getRow(self, rowIndex):
        """ :type rowIndex: int :rtype: List[int] """
        res = [1] + [0] * rowIndex
        for i in range(rowIndex):
            res[0] = 1
            for j in range(i+1, 0, -1):
                res[j] = res[j] + res[j-1]
        return res

思路二

另外,也可以不考虑上一行,直接根据公式求出当前行的内容。帕斯卡三角形的第n行第m个元素是 C(n, m) = n!/(m! * (n-m)!)。所以 C(n, m-1) = n!/((m-1)! * (n-m+1)!),所以 C(n, m) = C(n, m-1) * (n-m+1) / m。另外,有组合数的性质可以减少一半计算量:C(n, m) == C(n, n-m)。

代码

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

PS: 新手刷LeetCode,新手写博客,写错了或者写的不清楚还请帮忙指出,谢谢!
转载请注明:http://blog.csdn.net/coder_orz/article/details/51591374

    原文作者:杨辉三角问题
    原文地址: https://blog.csdn.net/coder_orz/article/details/51591374
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞