119.Pascal’s Triangle II
Given an index k, return the kth row of the Pascal’s triangle.
For example, given k = 3,
Return [1,3,3,1].
Note:
Could you optimize your algorithm to use only O(k) extra space?
做过118. Pascal’s Triangle这道题的同学应该可以用同样的方法来做这道题。
方法一:
杨辉三角形的规律:每一行可以由上一行变形得到,先在上一行最前面插入1,然后从第二位开始,将每一位重置为该位与后面相邻一位之和,直到倒数第二位。
这里的row是存储型的,每次变换 row ,每次在最前面插入1,然后第二位到倒数第二位重置为相邻两数之和.得到新的一行。
且这种方法满足算法使用的额外空间是O(k)。
public List<Integer> getRow(int rowIndex) {
List<Integer> row = new ArrayList<>();
for(int i = 0; i <= rowIndex; i++){
row.add(0, 1);
for(int j = 1; j < row.size() - 1; j++){
row.set(j, row.get(j) + row.get(j + 1));
}
}
return row;
}
方法二:
使用一个List 来记录上一行的值。
public List<Integer> getRow2(int rowIndex) {
List<Integer> result = new ArrayList<>();
List<Integer> row = null;
for(int i = 0; i <= rowIndex; i++){
row = new ArrayList<>();
for(int j = 0; j <= i; j++){
if(j == 0 || j == i){
row.add(1);
}
else{
row.add(result.get(j - 1) + result.get(j));
}
}
result = row;
}
return result;
}