2015年6月29日星期一

Pascal's Triangle II leetcode

Given an index k, return the kth row of the Pascal's triangle.
For example, given k = 3,
Return [1,3,3,1].
与Pascal's Triangle 思路相同 难度是控制空间为O(k), 所以就只在一个arraylist上面更新修改.
与I不同的是我们对于每一行从后往前扫 每个i res[i]等于res[i-1]+ res[i]
 如果从前往后扫的话res[i]被覆盖但是对于下一个res[i + 1]还需要再用刀之前的res[i] 而不是新的res[i]
时间O(n^2) 空间O(k)

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

没有评论:

发表评论