LeetCode热题100 杨辉三角

题目描述

给定一个非负整数 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，中间是f[i][j] = f[i - 1][j - 1] + f[i - 1][j]。

代码

class Solution {
public:
    vector<vector<int>> generate(int numRows) {
        vector<vector<int>>ans;
        for(int i = 0; i < numRows; ++i)
        {
            vector<int>res(i + 1, 1);
            for(int j = 1; j < i; ++j)
            {
                res[j] = ans[i- 1][j - 1] + ans[i - 1][j];
            }
            ans.push_back(res);
        }
        return ans;
    }
};