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