题目
给定一个非负整数 numRows
, 生成「杨辉三角」的前 numRows
行。
在「杨辉三角」中,每个数是它左上方和右上方的数的和。
输入: numRows = 5
输出: [[1],[1,1],[1,2,1],[1,3,3,1],[1,4,6,4,1]]
题目解析
思路一
首先我们先定义一个存储结果的数组,然后进行判断,如果行数小于1的情况下,直接返回空数组,其次我们认识到杨辉三角两端都是1,然后是其余数等于上一行相邻两个元素相加,所以我们这里定义两个变量循环,在每次循环出当前行的所有值,其中i表示行,在循环中声明一个当前行数组,在声明一个循环j,j表示列,因为正好j从0开始,每行列数=行数+1,所以循环条件也就是从0到i,在循环中在中间元素判断条件,处理一下每列元素为上一行相邻两元素之和,如果不是则两端元素直接写1最后当前行写入结果数组在进行返回
/**
* @param {number} numRows
* @return {number[][]}
*/
var generate = function(numRows) {
let result = []
if (numRows < 1) {
return result
}
for (let i= 0;i < numRows; i++) {
const subArr = []
for(let j = 0; j<= i; j++) {
if (j > 0 && j < i ) {
subArr.push(result[i-1][j-1] + result[i-1][j])
} else {
subArr.push(1)
}
}
result.push(subArr)
}
return result
};
思路二
因为这类似于斐波那契数列,所以我们还可以用以下方式实现,i代表行数,j代表列数,最后我们使用对象把值存起来,就无需计算多次了
/**
* @param {number} numRows
* @return {number[][]}
*/
var generate = function(numRows) {
function fn(i, j){
let r = (j==0 || i==j)? 1: (resMap[`${i-1},${j}`] || fn(i-1, j)) + (resMap[`${i-1},${j-1}`] || fn(i-1, j-1))
//
resMap[`${i},${j}`] = r
return r
}
let res = [], resMap = {}
for(let i=0;i<numRows;i++){
let a = []
for(let j=0;j<=i;j++) a.push(fn(i,j))
res.push(a)
}
return res
};