leetcode119 杨辉三角1和2（C++）

关于vector<vector<int>> 的初始化问题。

如定义一个：vector<vector<int>> tmp;

如果不初始化，直接使用：int i=0;tmp[i].push_back(0);会报错，这是因为该二维向量为空，寻找下标索引i第一行找不到，故而报错。
二维vector的初始化一般用vector<vector<int> >tmp(m, vector<int>(n));这里m和n分别为矩阵的行和列。

如果需要修改行数和列数，那么可以使用resize函数来调整行列规模，如

vector<vector<int> > res;
        res.resize(m);//m行
        for (int k = 0; k < m; ++k){
            res[k].resize(n);//每行为n列
        }

也可以使用push_back来进行行列调整

vector<vector<int>> res(3,vector<int>(4));//生成三行四列矩阵
res.push_back(vector<int> (10));//添加一行长度为10的数组
res[0].push_back(5);//在第一行末尾添加一个元素5,只修改单行的列规模

下面通过leetcode118.杨辉三角来练习二维vector和resize的使用。

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

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

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

 

C++代码：

class Solution {
public:
    vector<vector<int>> generate(int numRows) {
        vector<vector<int>> res;
        res.resize(numRows);
        for(int i=0;i<res.size();i++){
            res[i].resize(i+1,0);//每行有i+1列，且全部置0
            res[i][0]=1;res[i][res[i].size()-1]=1;
        }
        for(int i=0;i<res.size();i++){
            for(int j=0;j<res[i].size();j++){
                if(res[i][j]==0){
                    res[i][j]=res[i-1][j-1]+res[i-1][j];
                }
            }
        }
        return res;
    }
};

杨辉三角进阶版：

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

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

输入: 4
输出: [1,3,3,1]

算法  空间复杂度：O(k)

class Solution {
public:
    vector<int> getRow(int rowIndex) {
        vector<int> res(rowIndex+1, 0);
        if(rowIndex <0 )return res;
        res[0] = 1;
        for(int i=0;i<=rowIndex; i++) {
            int j = i;
            while(j>=1) 
                res[j--] += res[j-1];
        }
        return res;
    }
};