杨辉三角的解法

要求返回前n行的杨辉三角所有值

点击查看更多通信与专业知识

118. 杨辉三角

难度简单493

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

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

示例:

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

按照定义直接写就行。每进入下一行就重新增加一个对应个数的数组，并将第一个和最后一个置为1，其余位res[n][i] = res[n-1][i-1] + res[n-1][i]。由于对称，还可以对其进行优化。

 

// class Solution {
// public:
//     vector<vector<int>> generate(int numRows) {
//         vector<vector<int>> res;
//         if(numRows == 0) return res;
//         for(int n = 0; n < numRows; n++){
//             res.push_back(vector<int>(n+1));
//             res[n][0] = res[n][n] = 1;
//             for(int i = 1; i <= n/2; i++){//可以只计算前一半
//                 res[n][n-i] = res[n][i] = res[n-1][i-1] + res[n-1][i];
//             }
//         }
//         return res;
//     }
// };

class Solution {
public:
    vector<vector<int>> generate(int numRows) {
        vector<vector<int>> res;
        if(numRows == 0) return res;
        for(int n = 0; n < numRows; n++){
            res.push_back(vector<int>(n+1));
            res[n][0] = res[n][n] = 1;
            for(int i = 1; i < n; i++){//算整行的话就不能取到n
                res[n][i] = res[n-1][i-1] + res[n-1][i];
            }
        }
        return res;
    }
};