题目:
给定一个非负索引 k,其中 k ≤ 33,返回杨辉三角的第 k 行。
题目要求O(K)空间复杂度,即只开一个大小为k的数组。
解析:
题目要求只能开一个数组,则只能利用滚动的思想,每一行与上一行有关。在杨辉三角中,每个数是它左上方和右上方的数的和:
将三角转化为数组的话(空位补0):
1 0 0
1 1 0
1 2 1
每个数就是它上一行正上方和左上方的数的和。设置数组的初始值1 1,遍历数组将之后的每一行的值求出,填到数组中,再在末尾补个1。但是有个问题就是不可以从前往后遍历数组,因为这样的话会出现上一行的值被覆盖的情况,例如:
第一行:1 1
第二行:1 2 1
第三行:1 3 4 1
前两行没有问题,但到了第三行时,求出第二个数为1+2=3,将2覆盖了,第三个数就成了1+3=4,,出现了错误,后面都会发生同样的错误。所以应该从后往前遍历,先在末尾补个1,不会影响前边的值。代码如下:
public List<Integer> getRow(int rowIndex) {
List<Integer> result = new ArrayList<Integer>(rowIndex + 1);
result.add(1);
if (rowIndex == 0)
return result;
result.add(1);
if (rowIndex == 1)
return result;
for (int i = 2; i <= rowIndex; i++) {
result.add(1);
for (int j = i - 1; j > 0; j--)
result.set(j, result.get(j) + result.get(j - 1));
}
return result;
}