给定一个非负整数 numRows
,生成「杨辉三角」的前 numRows
行。
在「杨辉三角」中,每个数是它左上方和右上方的数的和。
示例 1:
输入: numRows = 5 输出: [[1],[1,1],[1,2,1],[1,3,3,1],[1,4,6,4,1]]
示例 2:
输入: numRows = 1 输出: [[1]]
思路
动态规划
代码
class Solution {
public:
vector<vector<int>> generate(int numRows) {
vector<vector<int>> res;
if(numRows == 0)
return res;
res.push_back({1});
vector<int> lasttmp;
for(int i=1; i<numRows; ++i)
{
vector<int>tmp;
for(int j=0; j<i+1; ++j)
{
if(j == 0 || j == i)
tmp.push_back(1);
else
tmp.push_back(lasttmp[j]+lasttmp[j-1]);
}
res.push_back(tmp);
lasttmp = tmp;
}
return res;
}
};