https://leetcode-cn.com/problems/pascals-triangle-ii/description/
给定一个非负索引 k,其中 k ≤ 33,返回杨辉三角的第 k 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。
示例:
输入: 3 输出: [1,3,3,1]
进阶:
你可以优化你的算法到 O(k) 空间复杂度吗?
class Solution {
public List<Integer> getRow(int rowIndex) {
Integer[] result = new Integer[rowIndex+1];
Arrays.fill(result, 0);
result[0] = 1;
for(int i = 1; i<result.length; i++) {
for(int j=i;j>0;j--) {
result[j] = result[j] + result[j-1];
}
}
return Arrays.asList(result);
}
}
思路是在O(k)的空间复杂度的限制下,在长度为k的数组内部,从杨辉三角的第一行开始依次计算到第k行的最终结果。
代码中i的值是第k行,j是第j个数。j从i开始,是因为第i行共有i+1个数字,从后往前计算,避免了第i-1行计算结果被覆盖丢失。
主要需要理解的是杨辉三角可以看成是一行行的数组
1
1 1
1 2 1
1 3 3 1
1 4 6 4 1
每一个数组都比上一个数组多一个数,第一位和最后一位都是1, 其余数字都是上一个数组对应位置和前一位置的数字的和 a[i][j] = a[i-1][j] + a[i-1][j-1]