题目描述:
给定一个非负索引 k,其中 k ≤ 33,返回杨辉三角的第 k 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。
示例:
输入: 3
输出: [1,3,3,1]
进阶:
你可以优化你的算法到 O(k) 空间复杂度吗?
方法1:
主要思路:
(1)由于只要输出指定的某一行,故只需保存当前行的前一行结果即可;
class Solution {
public:
vector<int> getRow(int rowIndex) {
//处理特殊情形
if(rowIndex==0){
return {1};
}
if(rowIndex==1){
return {1,1};
}
//初始值
vector<int> res(2,1);
//逐行更新结果
for(int i=2;i<=rowIndex;++i){
vector<int> tmp(i+1,1);
for(int j=1;j<i;++j){
tmp[j]=res[j-1]+res[j];
}
res.swap(tmp);
}
return res;
}
};