给定一个非负整数 numRows,生成「杨辉三角」的前 numRows 行。
在「杨辉三角」中,每个数是它左上方和右上方的数的和。
示例 1:
输入: numRows = 5
输出: [[1],[1,1],[1,2,1],[1,3,3,1],[1,4,6,4,1]]
示例 2:
输入: numRows = 1
输出: [[1]]
提示:
1 <= numRows <= 30
Related Topics 数组 动态规划
/**
* 思路:
* 动态规划,定义一个arr[numRows][numRows]数组,从左上角到右下角画一个对角线,对角线下面的就是一个杨辉三角。
* 1、初始化:arr[0][col] == 0 ,arr[row][row]==0
* 2、递推公式:arr[row][col] = arr[row-1][col-1] + arr[row-1][col]
*/
import java.util.ArrayList;
class Solution {
public List<List<Integer>> generate(int numRows) {
int[][] arr = new int[numRows][numRows];
List<List<Integer>> ans = new ArrayList<>();
for (int row = 0; row < numRows; row++) {
List<Integer> rowAns = new ArrayList<>();
for (int col = 0; col <= row; col++) {
if (col == 0 || row == col) {
arr[row][col] = 1;
} else {
arr[row][col] = arr[row - 1][col - 1] + arr[row - 1][col];
}
rowAns.add(arr[row][col]);
}
ans.add(rowAns);
}
return ans;
}
}