在Java中,这题需要返回一个List,借此机会来回顾一下List的相关知识。
为了使用List接口的方法,我们可以使用以下这四个类
List接口包括Collection接口的所有方法。 这是因为Collection是List的超级接口。
回到这题:
思路:我们需要用O(rowIndex)的空间复杂度进行解题,那么我们只能创建一个大小为rowIndex的List<Integer>。
如果给定的行数rowIndex小于2,则直接返回,否则需要进行迭代计算,每次迭代依赖“上一行”的数据,迭代之后的数据覆盖“上一行”的数据,每次迭代计算时,从后往前进行计算,可以避免“上一行”数据发生变化。
重点就是要从后往前计算
代码:
public static List<Integer> getRow(int rowIndex) {
List<Integer> list = new ArrayList<>(rowIndex);
for (int i = 0; i < rowIndex + 1; i++) {
list.add(1);
}
if(rowIndex < 2) return list;
for (int i = rowIndex - 1; i > 0; i--) {
for (int j = rowIndex - i; j > 0; j--) {
list.set(j, list.get(j) + list.get(j - 1));
}
}
return list;
}