题目描述(牛客网):
给出一个索引k,返回杨辉三角的第k行
例如,k=3,
返回[1,3,3,1].
备注:
你能将你的算法优化到只使用O(k)的额外空间吗?
Given an index k, return the k th row of the Pascal’s triangle.
For example, given k = 3,
Return[1,3,3,1].
Note:
Could you optimize your algorithm to use only O(k) extra space?
在Pascal三角形中,每个数字都是其正上方两个数字的和。
例:
Input: 3
Output: [1,3,3,1]
主要思想:
题意是指定行数,输出杨辉三角形的某行的元素值。
由于下一层比上一层多一个数,只需将上层的List增加1,然后向前做更新、覆盖即可,不用额外开辟新的存储空间;共用一个List,空间复杂度为O(k)。
实现代码:
import java.util.ArrayList;
public class Solution {
public ArrayList<Integer> getRow(int rowIndex) {
ArrayList<Integer> list = new ArrayList<>();
for (int i = 0; i <= rowIndex; i++) {
list.add(1);
for (int j = i - 1; j > 0; j--) {
list.set(j, list.get(j - 1) + list.get(j));
}
}
return list;
}
}
注意:只能第二重 for 只能向前做遍历;如果先后遍历做覆盖的话,则会使得后面使用的数并非原来的数,而是更新后的数。