这里写目录标题
- 题目描述:
- 输入:
- 输出:
- 代码实现:
题目描述:
给定一个非负整数 numRows,生成「杨辉三角」的前 numRows 行。
在「杨辉三角」中,每个数是它左上方和右上方的数的和。
输入:
numRows = 5
输出:
[[1],[1,1],[1,2,1],[1,3,3,1],[1,4,6,4,1]]
代码实现:
import java.util.ArrayList;
import java.util.List;
public class Main{
public static void main(String[] args) {
List<List<Integer>> lists = generate(5);//调用函数,得到二维动态数组
System.out.println(lists);//[[1], [1, 1], [1, 2, 1], [1, 3, 3, 1], [1, 4, 6, 4, 1]]
}
public static List<List<Integer>> generate(int numRows) {
//创建一个二维数组:记录杨辉三角矩阵每个点的值
int[][] arr = new int[numRows][numRows + 1];//第一列全为0,所以二维数组要多一列
arr[0][1] = 1;//初始化第一行的值
//遍历数组,得到杨辉三角
for (int i = 1; i < arr.length; i++) {
for (int j = 0; j < arr[i].length - 1; j++) {
//计算逻辑:上一行不同时为零的两个数相加之和 = 当前所在行 且 竖直方向在后一个加数的正下方的点
int num1 = arr[i - 1][j];//第一个加数
int num2 = arr[i - 1][j + 1];//第二个加数
if (num1 != 0 || num2 != 0) {
//判断条件:不同时为零
arr[i][j + 1] = num1 + num2;
}
}
}
//创建二维动态数组
List<List<Integer>> lists = new ArrayList<>();
//遍历杨辉三角矩阵:将矩阵每一行存入一个一维动态数组,再存入二维数组
for (int i = 0; i < arr.length; i++) {
List<Integer> list = new ArrayList<>();//创建一维数组
for (int j = 0; j < arr[i].length; j++) {
if (arr[i][j] != 0) {
list.add(arr[i][j]);//将每一行不为零的数 存入一维动态数组
}
}
//再存入二维动态数组
lists.add(list);
}
return lists;//循环结束,最终返回二维动态数组
}
}