LeetCode-119. 杨辉三角 II

题目

给定一个非负索引 k，其中 k ≤ 33，返回杨辉三角的第 k 行。

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

示例:

输入: 3
输出: [1,3,3,1]

进阶：

你可以优化你的算法到 O(k) 空间复杂度吗？

解题

这题使用滚动数组来进行求解, 就拿上面的题目来说, 将其转换成数组来表示如下, 其中没有元素的地方在后面补0:

行数	数组值
第一行	1, 0, 0, 0, 0
第二行	1, 1, 0, 0, 0
第三行	1, 2, 1, 0, 0
第四行	1, 3, 3, 1, 0
第五行	1, 4, 6, 4, 1

• 可以看出, 每行的第i个元素, 就等于上一行的第[i]个元素与第[i - 1]个元素之和, 而且每行的第一个元素和最后一个元素都为1, 根据这个特点就可以从上往下依次求出每行的值了
• 对每一行进行遍历求值的时候需要注意, 要从后往前遍历, 不然会导致上一行的值被新行的值所覆盖, 而导致计算结果错误, 如针对第5行, 如果从前往后遍历的话, 会得到如下结果:

  1, 4, 7, 8, 1

• 还需要注意一点就是, 题目中输入的行数是从0开始的, 遍历的时候第一个元素和最后一个元素直接赋值为1即可, 代码如下:

class Solution {
    public List<Integer> getRow(int rowIndex) {
        Integer[] rows = new Integer[rowIndex + 1];
        rows[0] = 1;
        if(rowIndex > 0){
            rows[1] = 1;
        }

        for(int i = 2;i <= rowIndex; i++){
            for(int j = i - 1;j > 0;j--){
               rows[j] = rows[j] + rows[j - 1];
            }
            rows[i] = 1;
            
        }
        return Arrays.asList(rows);
    }
}