leetcode-数组-简单-杨辉三角

题目

给定一个非负整数 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;
    }
};