【LeetCode】119. 杨辉三角 II

119. 杨辉三角 II

难度:简单

题目

给定一个非负索引 rowIndex,返回「杨辉三角」的第 rowIndex 行。

在「杨辉三角」中,每个数是它左上方和右上方的数的和。

【LeetCode】119. 杨辉三角 II_第1张图片

示例 1:

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

示例 2:

输入: rowIndex = 0
输出: [1]

示例 3:

输入: rowIndex = 1
输出: [1,1]

提示:

  • 0 <= rowIndex <= 33

进阶:

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

个人题解

方法一:暴力枚举

思路:构建杨辉三角,直接返回对应层集合

class Solution {
    private static final int MAX_INDEX = 33;

    private static final List<List<Integer>> YANG_HUI_TRIANGLE = new ArrayList<>();

    static {
        for (int i = 0; i <= MAX_INDEX; i++) {
            List<Integer> list = new ArrayList<>();
            for (int j = 0; j < i + 1; j++) {
                if (j == 0 || j == i) {
                    list.add(1);
                } else {
                    list.add(YANG_HUI_TRIANGLE.get(i - 1).get(j - 1) + YANG_HUI_TRIANGLE.get(i - 1).get(j));
                }
            }
            YANG_HUI_TRIANGLE.add(list);
        }
    }

    public List<Integer> getRow(int rowIndex) {
        return YANG_HUI_TRIANGLE.get(rowIndex);
    }
}

官方题解:

方法一:递推

class Solution {
    public List<Integer> getRow(int rowIndex) {
        List<Integer> row = new ArrayList<Integer>();
        row.add(1);
        for (int i = 1; i <= rowIndex; ++i) {
            row.add(0);
            for (int j = i; j > 0; --j) {
                row.set(j, row.get(j) + row.get(j - 1));
            }
        }
        return row;
    }
}

方法二:线性递推

class Solution {
    public List<Integer> getRow(int rowIndex) {
        List<Integer> row = new ArrayList<Integer>();
        row.add(1);
        for (int i = 1; i <= rowIndex; ++i) {
            row.add((int) ((long) row.get(i - 1) * (rowIndex - i + 1) / i));
        }
        return row;
    }
}

作者:力扣官方题解
链接:https://leetcode.cn/problems/pascals-triangle-ii/solutions/601082/yang-hui-san-jiao-ii-by-leetcode-solutio-shuk/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

你可能感兴趣的:(LeetCode,leetcode,算法)