给定一个非负索引 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; }}