> 文档中心 > 力扣Java - 118. 杨辉三角

力扣Java - 118. 杨辉三角

杨辉三角

  • 题目描述
  • 示例演示
      • 示例一
      • 示例二
  • 算法思想
  • 算法代码
      • 方式一:返回List
      • 方式二:返回int型二维数组

题目描述

  • 一个非负整数 numRows,生成「杨辉三角」的前 numRows 行。
  • 「杨辉三角」中,每个数是它左上方和右上方的数的和。

在这里插入图片描述

示例演示

示例一

输入: numRows = 5
输出: [[1],[1,1],[1,2,1],[1,3,3,1],[1,4,6,4,1]]

示例二

输入: numRows = 1
输出: [[1]]

算法思想

  • 杨辉三角第 i 行具有第 i 个元素,头和尾的元素值是 1 。
  • 以二维数组 nums 为例解释。
  • 给首尾元素赋值:nums[i][0] = nums[i][i] = 1
  • 给中间元素赋值:nums[i][j] = nums[i-1][j] + nums[i-1][j-1]

算法代码

方式一:返回List

class Solution {    public List<List<Integer>> generate(int numRows) { List<List<Integer>> yang = new ArrayList<List<Integer>>(); for (int i = 0; i < numRows; i++){     List<Integer> list = new ArrayList<Integer>();     for (int j = 0; j <= i ; j++){  if (j == 0 || j == i){      list.add(1);  }else{      int num = yang.get(i-1).get(j) + yang.get(i-1).get(j-1);      list.add(num);  }     }     yang.add(list); } return yang;    }}

方式二:返回int型二维数组

class Solution {    public int[][] generate(int numRows) { int[][] yang = new int[numRows][]; for (int i = 0; i < numRows; i++){     yang[i] = new int[i+1];     yang[i][0] = yang[i][i] = 1;     for (int j = 1; j < yang[i].length - 1; j++){  yang[i][j] = yang[i-1][j] + yang[i-1][j-1];     } } return yang;    }}