动态规划 (118杨辉三角)

给定一个非负整数 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;
    }
}
上一篇:牛客java面试问答118-126【面向对象】


下一篇:leetcode【简单】118、杨辉三角