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;
     }
 }