LeetCode 119. 杨辉三角 II
题目
给定一个非负索引 k,其中 k ≤ 33,返回杨辉三角的第 k 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。
示例:
输入: 3
输出: [1,3,3,1]
进阶:
你可以优化你的算法到 O(k) 空间复杂度吗?
分析
每次保存本次结果和上次结果即可。还可以用数学方法,杨辉三角每一列等于(x + y)^k的各项系数。
C++代码
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> res;
vector<int> last;
if(rowIndex == 0){
res.push_back(1);
return res;
}
res.push_back(1);
last = res;
for(int i = 1; i <= rowIndex; i++){
res.clear();
res.push_back(1);
for(int j = 0; j < i - 1; j++){
res.push_back(last[j] + last[j + 1]);
}
res.push_back(1);
last = res;
}
return res;
}
};