力扣:118. 杨辉三角
描述
给定一个非负整数 numRows,生成「杨辉三角」的前 numRows 行。
在「杨辉三角」中,每个数是它左上方和右上方的数的和。
示例 1:
输入: numRows = 5
输出: [[1],[1,1],[1,2,1],[1,3,3,1],[1,4,6,4,1]]
示例 2:
输入: numRows = 1
输出: [[1]]
提示:
1 <= numRows <= 30
数学
第二行开始,除了边上的数字固定为1外,中间的数字都可以用tri[i][j] = tri[i - 1][j] + tri[i - 1 ][j - 1];表示
cpp
#include<iostream>
#include<vector>
using namespace std;
class Solution{
public:
vector<vector<int>> generate(int numRows){
vector<vector<int>> tri(numRows);
for(int i = 0; i < numRows; i++){
tri[i].resize(i + 1);
tri[i][0] = tri[i][i] = 1;
for(int j = 1; j < i; ++j){
tri[i][j] = tri[i - 1][j] + tri[i - 1 ][j - 1];
}
}
return tri;
}
};
int main()
{
Solution solution;
int numRows = 5;
vector<vector<int>> result = solution.generate(numRows);
for (const auto& row : result) {
for (int num : row) {
cout << num << " ";
}
cout << endl;
}
return 0;
}