1 解题思想
首先注意下第一题:
Leetcode 118. Pascal’s Triangle 杨辉三角 解题报告
这题只要求输出第k层的数,最好优化到O(k)的空间复杂度。
其实杨辉三角本身就不需要那么多空间,计算当前层的数只需要有上一层的结果就好,所以用两个数组(or List)不停狡猾就可以。
2 原题
Given an index k, return the kth 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?
3 AC解
public class Solution {
public List<Integer> getRow(int rowIndex) {
ArrayList<Integer> last=new ArrayList<Integer>();
last.add(1);
if(rowIndex==0)
return last;
for(int i=1;i<=rowIndex;i++){
ArrayList<Integer> tmp=new ArrayList<Integer>();
tmp.add(1);
for(int j=1;j<i;j++){
tmp.add(last.get(j)+last.get(j-1));
}
tmp.add(1);
last=tmp;
}
return last;
}
}