leetcode 118 119 杨辉三角 c++

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

典型的动态规划问题，递归方程: x[i][j] = x[i-1][j] + x[i-1][j-1]

代码：

class Solution {
public:
    vector<vector<int>> generate(int numRows) {
        vector<vector<int>> ret;
        if (numRows == 0)
            return ret;
        ret.push_back(vector<int>(1, 1));
        if (numRows == 1)	//n =1的情况
            return ret;
        ret.push_back(vector<int>(2, 1));
        if (numRows == 2) //n=2的情况
            return ret;
        for (int i = 2; i < numRows; i++) {
            vector<int> temp(i + 1, 1);
            for (int j = 1; j < i; j++)
                temp[j] = ret[i - 1][j - 1] + ret[i - 1][j];
            ret.push_back(temp);
        }
        return ret;
    }
};

leetcode 119：给定一个非负索引 k，其中 k ≤ 33，返回杨辉三角的第 k 行。

这道题只要输出第k行，那我们在使用动态规划时就不需要存储前面的k-1行数据了，所以我们可以使用原地修改数据的方法，在开始时我们直接建立一维vector x(k+1,1)，在计算每一行的数据时我们要从最后一个数开始逆序修改，这样我们向前循环计算x[k]时，x[k]、x[k-1]此时还是前一行的数据，故当前行的x[k] = x[k]+x[k-1]，我们，这样我们就只要使用O(k)的额外空间就可以计算出第k行杨辉三角了。

代码：

vector<int> getRow(int rowIndex) {
        vector<int> ret(rowIndex + 1, 1);
        for (int i = 2; i <= rowIndex; i++) {
            for (int j = i - 1; j > 0; j--) 
                ret[j] = ret[j] + ret[j - 1];
        return ret;
    }