给定一个非负索引 k,其中 k ≤ 33,返回杨辉三角的第 k 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。
示例:
输入: 3
输出: [1,3,3,1]
进阶:
你可以优化你的算法到 O(k) 空间复杂度吗?
解析:本题与上一题的不同是仅需要输出一行,另外额外要求空间复杂度O(K)
因此我们仅设置一维数组,代表每一行的值,每次计算一行,然后用这一行的数值计算下一行并覆盖。
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> res(rowIndex+1);
if(rowIndex==0){
res[0]=1;
return res;
}
else{
vector<int> tmp = getRow(rowIndex-1);
for(int i=0;i<rowIndex+1;++i){
if(i==0||i==rowIndex) res[i]=1;
else res[i]=tmp[i-1]+tmp[i];
}
return res;
}
}
};