动态规划简单 NC245 杨辉三角(一)

NC245 杨辉三角(一)

描述

给定一个非负整数 num ，生成杨辉三角的前 num 行。
杨辉三角中，每个数是左上方和右上方的数之和。
数据范围： 1 \le num \le 30 \1≤num≤30
例如当输入为4时，对应的返回值为[[1],[1,1],[1,2,1],[1,3,3,1]]，打印结果如下图所示：

分析

每层两端都是1，其余都是根据上一层的计算得来。
想要返回的数组没有多余的0，实现[[1],[1,1]]而不是[[1,0],[1,1]，方法是每层都新建一个长度合适的数组，赋完值后，赋给二维数组。

import java.util.*;
public class Solution {
    public int[][] generate (int num) {
        int[][] dp = new int[num][num];
        for(int i = 0; i < num; i++){
            int[] tmp = new int[i+1];
            for(int j = 0; j <= i; j++){
                if(j == 0 || j == i){
                    tmp[j] = 1;
                }else{
                    tmp[j] = dp[i-1][j] + dp[i-1][j-1];
                }
            }
            dp[i] = tmp;
        }
        return dp;
    }
}