题目
给定一个非负整数 numRows,生成杨辉三角的前 numRows 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。
来源:https://leetcode-cn.com/problems/pascals-triangle/
输入: 5
输出:
[
[1],
[1,1],
[1,2,1],
[1,3,3,1],
[1,4,6,4,1]
]
解答
方法一:
我解答这个题目的时候分析是这样的:
- 假设从row是从0 开始的
- 除row = 0 之外,所有的行两边的元素都是1
- 每行又 row + 1 个元素
- 每行除了 index = 0 index = row 之外的元素是上一行的 index - 1 和 index 的元素之和, row= 1 除外
因此构建每行的vector的时候,直接全部初始化为1,然后再修改元素
class Solution {
public:
vector<vector<int>> generate(int numRows) {
vector<vector<int>> res;
if (numRows >= 1) {
res.push_back({1}); //row 0
}
if (numRows >= 2) {
res.push_back({1,1}); //row 1
}
for (int row = 2; row < numRows; row++) {
vector<int> tmp(row + 1, 1); // 共有row + 1个元素
for (int col = 1; col < row; col++) {
///上一行的元素相加得到
tmp[col] = res[row - 1][col - 1] + res[row - 1][col];
}
///元素构建完成
res.push_back(std::move(tmp));
}
return res;
}
};