[LeetCode] 118、杨辉三角

最新推荐文章于 2024-07-27 19:43:02 发布

原创最新推荐文章于 2024-07-27 19:43:02 发布 · 96 阅读

0 ·

CC 4.0 BY-SA版权

算法（UVa + LeetCode + OJ + ……）专栏收录该内容

442 篇文章

订阅专栏

本文介绍了一种使用C++编程语言生成杨辉三角的方法。通过详细解析算法思路，演示了如何根据给定的非负整数生成指定行数的杨辉三角，并提供了两种实现方式的代码示例。

题目描述

给定一个非负整数 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>> res;
        // 下面这一大坨都是特判
        if(numRows <= 0)
            return res;
        if(numRows == 1){
            vector<int> temp;
            temp.push_back(1);
            res.push_back(temp);
            return res;
        }
        vector<int> temp;
        temp.push_back(1), res.push_back(temp);
        temp.clear();
        temp.push_back(1), temp.push_back(1), res.push_back(temp);
        if(numRows == 2)
            return res;
        
        for(int i = 2; i < numRows; i++){
            temp.clear();
            temp.push_back(1);
            for(int j = 1; j < i; j++)
                temp.push_back(res[i-1][j-1] + res[i-1][j]);
            temp.push_back(1);
            res.push_back(temp);
        }
        
        return res;
    }
};

更简洁点的写法：

vector<vector<int>> generate(int numRows) {
    vector<vector<int>> ans;
    for(int i=0;i<numRows; ++i){
        ans.push_back(vector<int>());
        vector<int>& cols=ans[i];
        cols.push_back(1);
        for(int j=1;j<i;++j) cols.push_back(ans[i-1][j-1] + ans[i-1][j]);
        if(cols.size() < i+1) cols.push_back(1);
    }
    
    return ans;
}