LeetCode119. Pascal's Triangle II

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;
}