杨辉三角 II - 技术栈

杨辉三角 II

linsa_pursuer2024-10-26 11:24

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

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

示例 1:

复制代码

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

示例 2:

复制代码

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

示例 3:

复制代码

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

提示:

0 <= rowIndex <= 33

进阶：

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

java 复制代码

class Solution {
    public List<Integer> getRow(int rowIndex) {
        List<List<Integer>> list = new ArrayList<>();
        for (int i = 0; i <= rowIndex; i++) {
            List<Integer> temp = new ArrayList<>();
            if (i == 0) {
                temp.add(1);
            } else {
                List<Integer> last = list.get(i - 1);
                for (int j = 0; j < last.size(); j++) {
                    if (j == 0) {
                        temp.add(last.get(j));
                    }
                    if (j > 0) {
                        temp.add(last.get(j - 1) + last.get(j));
                    }
                    if (j == last.size() - 1) {
                        temp.add(last.get(last.size() - 1));
                    }
                }
            }
            list.add(temp);
        }
        return list.get(list.size() - 1);
    }
}