leetcode 118.杨辉三角

题干

给定一个非负整数 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> > yhTri(numRows);
        if(numRows == 0){
            return yhTri;
        }
        yhTri[0].push_back(1);
        for(int i = 1 ; i < numRows ; ++i){
            for(int j = 0 ; j < i + 1 ; ++j){
                yhTri[i].push_back(j == 0 || j == i ? 1 : yhTri[i-1][j-1] + yhTri[i-1][j]);
            }
        }
        return yhTri;
    }
};