100.杨辉三角II

一、题目描述

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

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

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

二、解题思路

2.1 思路一

先参考我的99_杨辉三角，求出整个杨辉三角，再获取指定的行即可。假设指定了需要第rowIndex行的元素，那么时间复杂度就是O(rowIndex^2/2)，因为是矩阵的下半边，只有一半，但系数一般都去掉所是O(rowIndex^2)。

思路一代码优化：

思路一时先将整个杨辉三角求出来，然后取所需要的那一行，但是发现所需的那一行都是依赖于前一行的值来计算的，所以不用先求出整个杨辉三角，每次只需要求出前一行，再根据前一行来计算所需的行即可。空间复杂度是O(rowIndex^2)因为每次都申请了k的空间，申请了rowIndex次。空间复杂度是相对小了一点。

2.2 思路二（进阶）：使用一个数组

假设输入的是5，需要杨辉三角的第5行元素：

先初始化一个长度为5的数组：

第一步：我们用这个数组来个更新每一行的元素，首先是第一行肯定就是1，用数组的第一个位置表示第一行：

第二步：现在求第二行的元素，第二行两个元素，首先知道第一个元素是1，第二个元素也是1，第二个元素1可以用它前一个元素也就是第一个元素1加上它本身位置的0获得：

第三步：接下来求第三行，第三行三个元素占数组的前三个位置，第三个位置是它前一个位置的值加上它本身位置的值来的也就是1，但是此时第三个位置前一个位置（第二个位置）的值是要更新的，这个值的更新等于第一个元素加上第二个元素（第二个位置本身），就相当于第二行的两个元素值相加。

第四步：再来求第四行，第四行第四个元素还是在第三行的基础上，用第三个位置的值加上第四个位置的值1+0=1：

然后来从右往左遍历更新元素，第三个值应该等于第三行（现在前三个元素就是第三行的杨辉三角值）的第二个元素加第三个元素，其实就是当前要更新位置的值加上当前位置的前一个元素。

继续往前遍历更新，还是当前位置的值加上前一个位置的值：

第五步：首先还是最后一个元素是1：

然后在从后往前比遍历，更新每个元素，当前要更新的元素等于它本身加上它前一个位置的元素：

至此使用一个数组求所需杨辉三角某行的值的模拟就结束了，下面看看代码实现。

三、代码演示

思路一：

class Solution {
    public List<Integer> getRow(int rowIndex) {
        List<List<Integer>> rows = new ArrayList<>();

        //对数组进行遍历
        for (int row=0; row<=rowIndex; row++){
            //当前行的元素集合
            List<Integer> oneRow = new ArrayList<>();
            for (int col=0; col<=row; col++){
                if (col==0 || col==row){
                    oneRow.add(1);
                }else{
                    List<Integer> preRow = rows.get(row-1);
                    oneRow.add(preRow.get(col-1)+preRow.get(col));
                }
            }
            //将每一行的元素放入到结果集中
            rows.add(oneRow);
        }
        return rows.get(rowIndex);
    }
}

思路一优化：

class Solution {
    public List<Integer> getRow(int rowIndex) {
        List<Integer> preRow = new ArrayList<>();

        //对数组进行遍历
        for (int row=0; row<=rowIndex; row++){
            //当前行的元素集合
            List<Integer> oneRow = new ArrayList<>();
            for (int col=0; col<=row; col++){
                if (col==0 || col==row){
                    oneRow.add(1);
                }else{
                    oneRow.add(preRow.get(col-1)+preRow.get(col));
                }
            }
            //将每一行的元素放入到结果集中
            preRow = oneRow;
        }
        return preRow;
    }
}

思路二（进阶）：使用一个数组

class Solution {
    public List<Integer> getRow(int rowIndex) {
        List<Integer> oneRow = new ArrayList<>();
        //第一个值肯定是1
        oneRow.add(1);
        for (int row=1; row<=rowIndex; row++) {
            //每多遍历一行就多申请一个空间，初始赋值为0即可
            oneRow.add(0);
            //逆序遍历
            for (int col=row; col>0; col--){
                //更新元素:当前位置的值更新等于它本身加上上一个位置的值
                oneRow.set(col, oneRow.get(col-1)+oneRow.get(col));
            }
        }
    return oneRow;
    }
}