118、给定一个非负整数 numRows,生成杨辉三角的前 numRows 行。 在杨辉三角中,每个数是它左上方和右上方的数的和。
示例:
输入: 5
输出: [
[1],
[1,1],
[1,2,1],
[1,3,3,1],
[1,4,6,4,1] ]
按照题意构建vector二维数组。难点在于,vec[0][0]的构建区别于之后几行的规律。找杨辉三角的规律,逐行求解。(i,j)的值为(i-1,j-1)的值加上(i-1,j)的值。
我先构建了一个大的二位数组vec,和一个一维数组tmp,每次先对某行的元素求解,完成tmp数组,插入大的二维数组vec中,下一次开始前对tmp进行清除。
class Solution {
public:
vector<vector<int>> generate(int numRows) {
vector<vector<int>> vec;
vector<int> tmp;
if(numRows==0) return vec;
tmp.push_back(1);
vec.push_back(tmp);
if(numRows==1) return vec;
for(int i=1;i<numRows;i++){
tmp.clear();
tmp.push_back(1);
for(int j=1;j<i;j++){
tmp.push_back(vec[i-1][j-1]+vec[i-1][j]);
}
tmp.push_back(1);
vec.push_back(tmp);
}
return vec;
}
};
执行用时 :4 ms, 在所有 C++ 提交中击败了71.29% 的用户
内存消耗 :8.8 MB, 在所有 C++ 提交中击败了40.13%的用户
修改1:不另外使用一维数组,一开始对vec进行初始化行数。
且将numRows的判断写入循环,将(0,0)的值的获取也巧妙的在第二次插入1前判断行是否为第一行,避免重复插入。
优雅的写法。
class Solution {
public:
vector<vector<int>> generate(int numRows) {
vector<vector<int>> vec(numRows);
for(int i=0;i<numRows;i++){
vec[i].push_back(1);
for(int j=1;j<i;j++){
vec[i].push_back(vec[i-1][j-1]+vec[i-1][j]);
}
if(i>0) vec[i].push_back(1);
}
return vec;
}
};
执行用时 :4 ms, 在所有 C++ 提交中击败了71.29% 的用户
内存消耗 :8.9 MB, 在所有 C++ 提交中击败了5.03%的用户
二维vector
创建二维数组: vector<vector<int>> vec();
创建有初始化行数的二维数组:vector<vector<int>> vec(numRows);
创建初始化行列的二维数组:
vector<vector<int>> vec(numRows);
for (i = 0; i < vec.size(); i++)
vec[i].resize(numCol);
访问某个元素:vec[i][j]
二维数组插入一个行元素:
vector<vector<int>> vec();
vector<int> tmp;
tmp.push_back(1); //tmp为[1]
vec.push_back(tmp); //vec为[[1]]