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

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

示例 1:

输入: rowIndex = 3
输出: [1,3,3,1]
示例 2:

输入: rowIndex = 0
输出: [1]
示例 3:

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

0 <= rowIndex <= 33
进阶:

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

Related Topics
数组
动态规划

方法一:保存整个杨辉三角形

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

方法二:使用滚动数组保存上一行

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

方法三:单数组

使用一个数组保存结果,每次组装下一行时,先给数组最后以为默认添加一个0,然后从后往前计算当前行,此时数组中保存的就都是上一行的结果

class Solution {    public List<Integer> getRow(int rowIndex) {        List<Integer> row = new ArrayList<>();        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 - 1) + row.get(j));            }        }        return row;    }}