1、题目
给定一个非负索引 rowIndex,返回「杨辉三角」的第 rowIndex 行。
在「杨辉三角」中,每个数是它左上方和右上方的数的和。
链接:https://leetcode-cn.com/problems/pascals-triangle-ii/
2、思路分析
1)暴力求解
将每行杨辉三角的计算出来,放在数组中,然后将最后一行,写入数组,最后返回。
缺点也很明显,就是空间复杂度很大,也是可以避免的。
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<vector<int>> nums(rowIndex + 1);
for(int i = 0; i < nums.size(); ++i){
nums[i].resize(i + 1, 1);
for(int j = 1; j < nums[i].size() - 1; ++j){
nums[i][j] = nums[i-1][j] + nums[i-1][j-1];
}
}
vector<int> ret;
for(int i = 0; i <= rowIndex; ++i){
ret.push_back(nums[rowIndex][i]);
}
return ret;
}
};
2)递推
每个数字等于上一行的左右两个数字之和,可用此性质写出整个杨辉三角。即第 n 行的第 i 个数等于第 n-1 行的第 i-1 个数和第 i 个数之和。
我们可以一行一行地计算杨辉三角。每当我们计算出第 i 行的值,我们就可以在线性时间复杂度内计算出第 i+1行的值。
class Solution {
public:
vector<int> getRow(int rowIndex) {
// 递推
vector<int> ret(rowIndex+1);
ret[0] = 1;
for(int i = 1; i <= rowIndex; ++i){
for(int j = i; j > 0; --j){
ret[j] += ret[j-1];
}
}
return ret;
}
};