118. 杨辉三角

问题

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

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

示例 1:
输入: numRows = 5
输出: [[1],[1,1],[1,2,1],[1,3,3,1],[1,4,6,4,1]]

分析

很容易想到递推。注意，memset只能用于0和-1，容易忘。

代码

class Solution {
public:
    int q[31][31];
    vector<vector<int> > res;
    vector<vector<int>> generate(int numRows) {
        int n_n = numRows;
        for (int i = 0; i < numRows; i++) { // 先初始化为1
            q[i][0] = 1;
            q[i][i] = 1;
        }
        // memset(q, 1, sizeof q); // 只能用于0和-1
        for (int i = 1; i < n_n; i++) {
            for (int j = 1; j < i; j++) {
                q[i][j] = q[i-1][j-1]+q[i-1][j]; // 递推
            }
        }
        // 构造输出格式
        for (int i = 0; i < numRows; i++) {
            vector<int> temp;
            for (int j = 0; j < i+1; j++) {
                temp.push_back(q[i][j]);
            }
            res.push_back(temp);
        }
        return res;
    }
};