给定索引 k,返回 Pascal 三角形的第 k 行。
例如,给定 k = 3,返回 [1,3,3,1]。
注意: 您可以优化算法以仅使用 O(k)额外空间吗?
解题思路:
- 与返回所有行不同的是,这里返回某一行的数据,同时要考虑到空间复杂度为 O(k),
所以应该想到在这一行数据里进行迭代。 - 利用 List 列表的 set 函数正好可以实现这一点,同时考虑到数据在迭代的过程中会出现覆盖的情况,所以从列表的后面向前迭代。
public ArrayList<Integer> getRow(int rowIndex) {
ArrayList<Integer> list = new ArrayList<>();
if(rowIndex < 0) return list;
rowIndex++;
list.add(1);
for(int i = 1; i < rowIndex; i++){
//i 执行的次数就表示要总的迭代几次
for(int j = i - 1; j > 0; j--){
list.set(j,list.get(j - 1) + list.get(j));
}
list.add(1);
}
return list;
}