leetcode 119. Pascal's Triangle II

Seawolf

1. Given a non-negative index k where k ≤ 33, return the kth index row of the Pascal's triangle.
   
2. 
   
3. Note that the row index starts from 0.
   
4. 
   
5. 
   
6. In Pascal's triangle, each number is the sum of the two numbers directly above it.
   
7. 
   
8. Example:
   
9. 
   
10. Input: 3
    
11. Output: [1,3,3,1]
    
12. Follow up:
    
13. 
    
14. Could you optimize your algorithm to use only O(k) extra space?

复制代码

1. class Solution {
   
2.     public List<Integer> getRow(int rowIndex) {
   
3.         
   
4.         List <Integer> array = new ArrayList<>();
   
5.         array.add(1);
   
6.         if(rowIndex == 0) return array;
   
7.         List <List<Integer>> re = new ArrayList<>();
   
8.         re.add(array);
   
9.         
   
10.         for(int i = 1; i <= rowIndex; i++){
    
11.             
    
12.             List<Integer> pre = re.get(i-1);
    
13.             List<Integer> cur = new ArrayList<>();
    
14.             cur.add(1);
    
15.             
    
16.             for(int j = 1; j < i ; j++){
    
17.                
    
18.                 cur.add(pre.get(j) + pre.get(j-1));
    
19.             }
    
20.             
    
21.             cur.add(1);
    
22.             re.add(cur);
    
23.         }
    
24.         
    
25.         return re.get(re.size()-1);
    
26.     }
    
27. }

复制代码

optimized with constant space

1. class Solution {
   
2.     public List<Integer> getRow(int rowIndex) {
   
3.         
   
4.         ArrayList <Integer> array = new ArrayList<>();
   
5.         array.add(1);
   
6.         if(rowIndex == 0) return array;
   
7.         ArrayList<Integer> cur = new ArrayList<>();;
   
8.         
   
9.         for(int i = 1; i <= rowIndex; i++){
   
10.             cur.clear();
    
11.             cur.add(1);
    
12.             
    
13.             for(int j = 1; j < i ; j++){
    
14.                
    
15.                 cur.add(array.get(j) + array.get(j-1));
    
16.             }
    
17.             
    
18.             cur.add(1);
    
19.             array.clear();
    
20.             array = (ArrayList<Integer>) cur.clone();
    
21.         }
    
22.         
    
23.         return cur;
    
24.     }
    
25. }

复制代码