【Leetcode065】杨辉三角II

119、杨辉三角II【简单】

给定一个非负索引 rowIndex，返回「杨辉三角」的第 rowIndex 行。

在「杨辉三角」中，每个数是它左上方和右上方的数的和。

1783c.png)

示例 1:
输入: rowIndex = 3
输出: [1,3,3,1]

示例 2:
输入: rowIndex = 0
输出: [1]

示例 3:
输入: rowIndex = 1
输出: [1,1]

提示:
0 <= rowIndex <= 33

进阶：
你可以优化你的算法到 O(rowIndex) 空间复杂度吗？

方法一：递推

1.1 思路分析

定义一个二维列表，存储每一行的数据，计算当前行时，向上一行查找。

1.2 代码实现

class Solution {
    public List<Integer> getRow(int rowIndex) {
        List<List<Integer>> C = new ArrayList<List<Integer>>(); // 记录总的
        for (int i=0; i<=rowIndex; i++){
            List<Integer> row = new ArrayList<Integer>();   // 记录每行
            for(int j=0; j<=i; j++){
                if (j==0 || j==i){
                    row.add(1);
                }else {
                    row.add(C.get(i-1).get(j-1) + C.get(i-1).get(j));
                }
            }
            C.add(row);
        }
        return C.get(rowIndex);
    }
}

1.3 测试结果

1.4 复杂度

• 时间复杂度：O(n²)
• 空间复杂度：O(n(n-1)/2)。杨辉三角的元素个数

方法二：递推优化v1.0

2.1 思路分析

因为只根据前一行计算当前行，所以将二维列表换成一维列表pre，用于记录前一行数据。记录完当前行，便把pre变成当前行cur.

2.2 代码实现

class Solution {
    public List<Integer> getRow(int rowIndex) {
        List<Integer> pre = new ArrayList<Integer>();   // 记录前一行
        for (int i=0; i<=rowIndex; i++){
            List<Integer> cur = new ArrayList<Integer>();   // 记录当前行
            for(int j=0; j<=i; j++){
                if (j==0 || j==i){
                    cur.add(1);
                }else {
                    cur.add(pre.get(j-1) + pre.get(j));
                }
            }
            pre = cur;
        }
        return pre;
    }
}

2.3 测试结果

2.4 复杂度

• 时间复杂度：O(n²)
• 空间复杂度：O(rowIndex)

方法三：递推优化v2.0

3.1 思路分析

递推式${\mathcal{C}}_n^i={\mathcal{C}}_{n-1}^i+{\mathcal{C}}_{n-1}^{i-1}$

​表明，当前行第 i 项的计算只与上一行第 i−1 项及第 i 项有关。因此我们可以倒着计算当前行，这样计算到第 i 项时，第 i−1 项仍然是上一行的值。

3.2 代码实现

class Solution {
    public List<Integer> getRow(int rowIndex) {
        List<Integer> row = new ArrayList<Integer>();   // 记录前一行
        row.add(1);
        for (int i=1; i<=rowIndex; ++i){
            row.add(0);
            for(int j=i; j>0; --j){
                row.set(j, row.get(j) + row.get(j-1));
            }
        }
        return row;
    }
}

3.3 测试结果

3.4 复杂度

• 时间复杂度：O(rowIndex²）
• 空间复杂度：O(1)。不考虑返回值的空间占用。

方法四：线性递推

4.1 思路分析

4.2 代码实现

class Solution {
    public List<Integer> getRow(int rowIndex) {
        List<Integer> row = new ArrayList<Integer>();
        row.add(1);
        for (int i = 1; i <= rowIndex; ++i) {
            row.add((int) ((long) row.get(i - 1) * (rowIndex - i + 1) / i));
        }
        return row;
    }
}

4.3 测试结果

4.4 复杂度

• 时间复杂度：O(rowIndex²）
• 空间复杂度：O(1)。不考虑返回值的空间占用。