给定一个非负整数 numRows,生成杨辉三角的前 numRows 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。
示例:
输入: 5
输出:
[
[1],
[1,1], [1,2,1], [1,3,3,1], [1,4,6,4,1]
]
代码:
class Solution {
public:
vector<vector<int>> generate(int numRows) {
vector<int> v;
vector<int> v1;
vector<vector<int>> v2;
if(numRows==0){
v2.push_back(v1);
return v2;
}
v1.push_back(1);
//补充0的,用来计算的一维数组
v.push_back(0);
v.push_back(1);
v.push_back(0);
v2.push_back(v1);
int start=0;
for(int i=2;i<=numRows;i++){
vector<int> nv; //当前层的数组
for(int j=0;j<i;j++){ //第i层的元素个数是i,进行i次运算
nv.push_back(v[start+j]+v[start+j+1]);
}
v.insert(v.end(),nv.begin(),nv.end());
v.push_back(0); //在运算数组后面补0
start=start+i;
v2.push_back(nv);
}
return v2;
}
};
根据图示找规律:
①第i层有i个元素。
②每一层左起第一个和最后一个元素都是1
②下一层元素是上一层v[i]+v[i+1]。每层第一个元素1不符合,需要每次计算下一层前,在当前层后补个0。
具体如图: