题目
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?
翻译
假设有一个索引k,返回帕斯卡三角(杨辉三角)第k行。比如:
给定索引k。则返回【1,3,3,1】
注:
是否可以优化空间复杂度为O(k)
思路
我们来模拟下:
0,【1】
1,【1,1】
2,【1,2,1】
3,【1,3,3,1】
即第i行和第i-1行的关系为
a[i][j] = a[i-1][j-1] + a[i-1][j] 。开头和结尾处为1。不用计算。
代码
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> row;
if(rowIndex < 0){
return row;
}
for(int i = 1; i <= rowIndex + 1; ++i){
vector<int> buf;
for(int j = 0; j < i; j ++){
if(i <= 2){
buf.push_back(1);
}else{
if(j > 0 && j < i-1) {
int tmp = row[j-1] + row[j];
buf.push_back(tmp);
}else{
buf.push_back(1);
}
}
}
row = buf;
}
return row;
}
};