题目
给定一个非负整数 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 };