力扣 119. 杨辉三角 II

题目

给定一个非负索引 k,其中 k ≤ 33,返回杨辉三角的第 k 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。

示例

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

你可以优化你的算法到 O(k) 空间复杂度吗?

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/pascals-triangle-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

方法1:滚动数组

1、将杨辉三角(到题目输入的数为止)存入一个列表list()中
2、用enumerate()函数遍历杨辉三角,再输出。

3、滚动数组实现思路
力扣 119. 杨辉三角 II_第1张图片

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

作者:fuxuemingzhu
链接:https://leetcode-cn.com/problems/pascals-triangle-ii/solution/gun-dong-shu-zu-de-dong-hua-tu-jie-you-h-rw5n/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
方法2:暴力
Python实现
class Solution:
    def getRow(self, rowIndex: int) -> List[int]:
        out=list()
        for i in range(rowIndex+1):
            var=list()
            for j in range(i+1):
                if j==0 or j==i:
                    var.append(1)
                else:
                    var.append(out[i-1][j-1]+out[i-1][j])
            out.append(var)
        return out[-1]

力扣 119. 杨辉三角 II_第2张图片

Java实现
class Solution {
    public List<Integer> getRow(int rowIndex) {
        List<List<Integer>> list = new ArrayList<>();
        for (int i = 0; i <= rowIndex; i++) {
            // List类存储每行的值
            List<Integer> row = new ArrayList<>();
            for (int j = 0; j <= i; j++) {
                // 边界值赋1
                if (j == 0 || j == i) {
                    row.add(1);
                } else {
                    row.add(list.get(i - 1).get(j - 1) + list.get(i - 1).get(j));
                }
            }
            list.add(row);
        }

        return list.get(rowIndex);
    }
}

在这里插入图片描述

你可能感兴趣的:(力扣,leetcode,算法,职场和发展)