LeetCode119：杨辉三角 II解析

最新推荐文章于 2022-02-05 00:08:30 发布

哪得小师弟

最新推荐文章于 2022-02-05 00:08:30 发布

阅读量278

点赞数

分类专栏： Leetcode算法题分析

本文链接：https://blog.csdn.net/x603560617/article/details/83749605

版权

Leetcode算法题分析专栏收录该内容

156 篇文章 2 订阅

订阅专栏

给定一个非负索引 k，其中 k ≤ 33，返回杨辉三角的第 k 行。

在杨辉三角中，每个数是它左上方和右上方的数的和。
示例：

输入: 3
输出: [1,3,3,1]

进阶：
你可以优化你的算法到 O(k) 空间复杂度吗？

这个题和118题没什么太大的区别，只是希望可以优化一下空间复杂度。也就是说之前是要存储整个三角，而现在只存之前的那一行，这个稍微改一点点就可以，其实就是动态规划的思想。
C++源代码：

class Solution {
public:
    vector<int> getRow(int rowIndex) {
        vector<int> pre;
        pre.push_back(1);
        if (rowIndex==0) return pre;
        pre.push_back(1);
        if (rowIndex==1) return pre;
        for (int i=2;i<=rowIndex;i++)
        {
            vector<int> now;
            now.push_back(1);
            for (int j=1;j<i;j++)
                now.push_back(pre[j-1]+pre[j]);
            now.push_back(1);
            pre = now;
        }
        return pre;
    }
};

python3源代码：

class Solution:
    def getRow(self, rowIndex):
        """
        :type rowIndex: int
        :rtype: List[int]
        """
        l = [1]
        if rowIndex==0:
            return l
        l.append(1)
        if rowIndex==1:
            return l
        for i in range(2, rowIndex+1):
            temp = [1]
            for j in range(1, i):
                temp.append(l[j-1]+l[j])
            temp.append(1)
            l = temp
        return l