[LeetCode]119. Pascal's Triangle II

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?

思路:之前的方法是利用动态规划求出杨辉三角的每一层，最后返回最后一层

改进后，不存放每一层的结果，直接用新的结果replace原来的结果，则符合对空间复杂度的要求

改进前：

public class Solution {
    public List<Integer> getRow(int rowIndex) {
    	List<List<Integer>> res =new ArrayList<List<Integer>>(rowIndex+1);
        	List<Integer> r = new ArrayList<Integer>();
        	r.add(1);
        	res.add(r);
        	for (int i = 1; i < rowIndex+1;i++){
        	    List<Integer> rt = new ArrayList<Integer>(i+1);
        		rt.add(1);
        		for (int j = 1;j < i;j++) {
        		rt.add(res.get(i-1).get(j-1)+res.get(i-1).get(j));	
        		}
        		rt.add(1);
        		res.add(rt);
        	} 
        return res.get(rowIndex);
    }
}

改进后：

 public List<Integer> getRow(int rowIndex) {
        List<Integer> res = new ArrayList<>();
        for(int i = 0; i <= rowIndex; i++) {
            res.add(1);
            for(int j = i-1; j > 0; j--) {
                res.set(j, res.get(j-1) + res.get(j));
            }
        }
        return res;
    }