leetcode 118.杨辉三角

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) {
        // 三种情况，当行数小于等于2时，直接返回相应的结果
        if(numRows <= 0){
            return {};
        }
        if (numRows == 1){
            return {{1}};
        }
        if (numRows == 2){
            return {{1},{1,1}};
        }
		
        // 当函数大于2时，需要初始化存储的变量，根据上一个的数组计算当前数组。
        vector<vector<int>> res={{1},{1,1}};
        vector<int> arr;
        int row = 1;
        while(row < numRows-1){
            arr.push_back(1);
            for(int i=0; i<res[row].size()-1; i++){
                int temp = 0;
                temp = res[row][i] + res[row][i+1];
                arr.push_back(temp);
            }
            arr.push_back(1);
            res.push_back(arr);
            arr.clear();
            row++;
        }
        return res;
    }
};

欢迎大家关注我的个人公众号，同样的也是和该博客账号一样，专注分享技术问题，我们一起学习进步