LeetCode 118杨辉三角
-
题目简述:给定一个非负整数
numRows
,生成杨辉三角的前numRows
行。在杨辉三角中,每个数是它左上方和右上方的数的和。 -
输入:5 输出:
[ [1], [1,1], [1,2,1], [1,3,3,1], [1,4,6,4,1] ]
-
思路:递推,时间复杂度O(n2)
- 定义答案,用第一行的
1
初始化,然后从上倒下依次计算每一行 - 首先将要计算的当前每一行首部的
1
单独加入数组,然后依次利用上一行的数组计算中间的数last[i] + last[i + 1]
,最后将当前行尾部的1
单独加入数组 - 将计算得到的当前行数组加入答案并返回
- 定义答案,用第一行的
class Solution {
public:
vector<vector<int>> generate(int numRows) {
vector<vector<int>> res;
if(numRows == 0) return res;
res.push_back(vector<int>(1, 1));//先加入第一行的一个1
for(int i = 1; i < numRows; i++)
{
vector<int> &last = res.back();
vector<int> cur;
cur.push_back(1);
for(int i = 0; i + 1 < last.size(); i++)//要约束i+1
cur.push_back(last[i] + last[i + 1]);
cur.push_back(1);
res.push_back(cur);
}
return res;
}
};