【leetcode】118 杨辉三角（迭代）

题目链接：https://leetcode-cn.com/problems/pascals-triangle/

题目描述

给定一个非负整数 numRows，生成杨辉三角的前 numRows 行。

在杨辉三角中，每个数是它左上方和右上方的数的和。

示例:

输入: 5
输出:
[
     [1],
    [1,1],
   [1,2,1],
  [1,3,3,1],
 [1,4,6,4,1]
]

代码

class Solution {
public:
    vector<vector<int>> generate(int numRows) {
        vector<vector<int>> ret;
        if(numRows < 1) return ret;
        ret.push_back({1});
        if(numRows == 1) return ret;
        ret.push_back({1,1});
        if(numRows == 2) return ret;
        for (int i = 3; i <= numRows; ++i) {
            vector<int> last = ret.back();
            vector<int> tmp = {1};
            for (int j = 1; j < last.size(); ++j)
                tmp.push_back(last[j]+last[j-1]);
            tmp.push_back(1);
            ret.push_back(tmp);
        }
        return ret;
    }
};