题目
给定一个非负索引 rowIndex
,返回「杨辉三角」的第 rowIndex
行。
思路
1、这道题和我之前发布的杨辉三角形几乎来说是一模一样的(具体可以看我的上一篇博客)
2、根据题目的提示要求使用O(rowIndex)空间复杂度,在创建杨辉三角形时,我们可以在二维数组中只创建rowIndex这么多行的一维数组
3、然后返回最后一行也就是(rowIndex行)的一维数组就行了
代码
class Solution {
public List<Integer> getRow(int rowIndex) {
if (rowIndex == 0) {
return new ArrayList<Integer>(List.of(1));
}
List<List<Integer>> c = new ArrayList<>(rowIndex+1);
c.add(List.of(1));
for (int i=1;i<rowIndex+1;i++) { 外循环作为从上到下依次数到要求的那一行
List<Integer> row = new ArrayList<>(i+1);
row.add(1);
for (int j=1;j<i;j++) { //内循环则是根据上一行的数据来计算本行的数据
row.add(c.get(i-1).get(j-1)+c.get(i-1).get(j));
}
row.add(1);
c.add(row);
}
return c.get(rowIndex);
}
}