来源

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/pascals-triangle/

描述

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

示例:
输入: 5
输出:
[
[1],
[1,1],
[1,2,1],
[1,3,3,1],
[1,4,6,4,1]
]

题解

使用动态规划,基于前一行来构造每一行。

  1. class Solution {
  2. public List<List<Integer>> generate(int numRows) {
  3. List<List<Integer>> triangle = new ArrayList<List<Integer>>();
  4. if (0 == numRows) return triangle;
  5. // first row is always [1].
  6. triangle.add(new ArrayList<Integer>());
  7. triangle.get(0).add(1);
  8. for (int rowNum = 1; rowNum < numRows; rowNum++) {
  9. List<Integer> row = new ArrayList<>();
  10. List<Integer> prevRow = triangle.get(rowNum - 1);
  11. // The first row element is always 1.
  12. row.add(1);
  13. for (int j = 1; j < rowNum; j++) {
  14. row.add(prevRow.get(j - 1) + prevRow.get(j));
  15. }
  16. // The last row element is always 1.
  17. row.add(1);
  18. triangle.add(row);
  19. }
  20. return triangle;
  21. }
  22. }

复杂度分析

  • 时间复杂度:118. 杨辉三角(Pascal's Triangle) - 图1
  • 空间复杂度:118. 杨辉三角(Pascal's Triangle) - 图2,因为我们需要存储我们在triangle中更新的每个数字,所以空间复杂度与时间复杂度相同。