给定一个非负索引 rowIndex,返回「杨辉三角」的第 rowIndex 行。
在「杨辉三角」中,每个数是它左上方和右上方的数的和。
示例 1:
输入: rowIndex = 3
输出: [1,3,3,1]
示例 2:
输入: rowIndex = 0
输出: [1]
示例 3:
输入: rowIndex = 1
输出: [1,1]
提示:
0 <= rowIndex <= 33
来源:力扣(LeetCode)
题目链接:https://leetcode-cn.com/problems/pascals-triangle-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题方法一:直接建立杨辉三角寻找
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<vector<int>> ans;
for(int i = 0; i <= rowIndex; ++i){
vector<int> row;
for(int j = 0; j <= i; ++j){
if(j == 0 || j == i) row.push_back(1);
else row.push_back(ans[i - 1][j - 1] + ans[i - 1][j]);
}
ans.push_back(row);
}
return ans[rowIndex];
}
};
解题方法二:空间优化使用preRow数组
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> row;
for(int i = 0; i <= rowIndex; ++i){
vector<int> preRow = row;
row.clear();
for(int j = 0; j <= i; ++j){
if(j == 0 || j == i) row.push_back(1);
else row.push_back(preRow[j - 1] + preRow[j]);
}
}
return row;
}
};
解题方法三:只用一个数组
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> row(rowIndex + 1);
for(int i = 0; i <= rowIndex; ++i){
for(int j = i; j >= 0; --j){
if(j == 0 || j == i) row[j] = 1;
else row[j] = row[j - 1] + row[j];
}
}
return row;
}
};