[LeetCode] 118. Pascal's Triangle

Given a non-negative integer numRows, generate the first numRows of Pascal's triangle.

In Pascal's triangle, each number is the sum of the two numbers directly above it.

Example 1:

Input: numRows = 5
Output: [[1],[1,1],[1,2,1],[1,3,3,1],[1,4,6,4,1]]

Example 2:

Input: numRows = 1
Output: [[1]]

Constraints:

• 1 <= numRows <= 30

杨辉三角。

给定一个非负整数 numRows，生成杨辉三角的前 numRows 行。

这个三角形注意一下几点

• 首行只有一个元素 1
• 每一行的第一个和最后一个元素是 1
• 中间行的元素个数 = 行的index。注意 index 是从 1 开始的
• 中间行位于index j 的元素 = 上一行位于 j - 1 的元素 + 上一行位于 j 的元素

时间O(n^2)

空间O(n^2) - list of list

Java实现

class Solution {
    public List<List<Integer>> generate(int numRows) {
        List<List<Integer>> res = new ArrayList<>();
        List<Integer> firstRow = new ArrayList<>();
        firstRow.add(1);
        res.add(firstRow);
        for (int i = 1; i < numRows; i++) {
            List<Integer> cur = new ArrayList<>();
            // 每一行开头的1
            cur.add(1);
            // 从第二个数字开始，都是找上一行的元素
            // 当前行位置 j 上的数字 = 上一行位置 j - 1 + 上一行位置 j 上的数字
            for (int j = 1; j < i; j++) {
                cur.add(res.get(i - 1).get(j - 1) + res.get(i - 1).get(j));
            }
            // 记得加每行最后一个1
            cur.add(1);
            res.add(cur);
        }
        return res;
    }
}

 

另一种Java实现，有助于版本二

思路是每次先往 list 的头部加一个 1，然后 j 位置是由 j 和 j + 1 位置上的数字的加和决定的，这样就能做到了重复利用 list

class Solution {
    public List<List<Integer>> generate(int numRows) {
        List<List<Integer>> res = new ArrayList<>();
        List<Integer> list = new ArrayList<>();
        for (int i = 0; i < numRows; i++) {
            list.add(0, 1);
            for (int j = 1; j < list.size() - 1; j++) {
                list.set(j, list.get(j) + list.get(j + 1));
            }
            res.add(new ArrayList<>(list));
        }
        return res;
    }
}

 

相关题目

118. Pascal's Triangle

119. Pascal's Triangle II

120. Triangle

799. Champagne Tower

2221. Find Triangular Sum of an Array

LeetCode 题目总结