LeetCode OJ 119. Pascal's Triangle II

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?

---

【思路】

一刷：我们为了满足空间复杂度的要求，我们新建两个ArrayList，一个负责存储上一个Pascal行的结果，一个根据上一个Pascal行得出当前Pascal行的结果，这个过程循环进行。

二刷：ArrayList的操作是比较慢的，这次我们用数组来对结果进行存储。数组保存上一个状态，通过上一个状态计算下一个状态。例如：[1,3,3,1]，我们可以从最后一个元素开始，num[i] = num[i] + num[i-1]。最后我们再在末尾补1。这样能大大提高代码的效率。

---

【java代码一刷】

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

 

---

【java代码二刷】

public class Solution {
    public List<Integer> getRow(int rowIndex) {
        int[] nums = new int[rowIndex+1];
        for(int i = 0; i <= rowIndex; i++) {
            for(int j = i; j >= 1; j--)
                nums[j] += nums[j-1];
            nums[i] = 1;
        }
        
        List<Integer> res = new ArrayList<>(rowIndex);
        for(int num : nums) res.add(num);
        return res;
    }
}