题目描述:
给定一个非负整数 numRows,生成杨辉三角的前 numRows 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。
示例:
输入: 5
输出:
[
[1],
[1,1],
[1,2,1],
[1,3,3,1],
[1,4,6,4,1]
]
方法1:
主要思路:
(1)直观的想,就是使用上一行的值相加,等于下一行的值;
(2)初始化前两行之后,后面的迭代更新即可;
class Solution {
public:
vector<vector<int>> generate(int numRows) {
//处理输入行是0,1,2的特殊情形
if(numRows==0){
return vector<vector<int>>();
}
if(numRows==1){
return vector<vector<int>>(1,vector<int>(1,1));
}
if(numRows==2){
vector<vector<int>> res;
vector<int> tmp1(1,1);
vector<int> tmp2(2,1);
res.push_back(tmp1);
res.push_back(tmp2);
return res;
}
//初始化杨辉三角前两行
vector<vector<int>> res;
vector<int> tmp1(1,1);
vector<int> tmp2(2,1);
res.push_back(tmp1);
res.push_back(tmp2);
//迭代更新随后的行
for(int i=1;i<numRows-1;++i){
vector<int> tmp(i+2,1);
for(int j=1;j<=i;++j){
tmp[j]=res[i][j-1]+res[i][j];
}
res.push_back(tmp);
}
return res;
}
};