LeetCode 119: Pascal's Triangle II

題目要求:https://leetcode.com/problems/pascals-triangle-ii/

1)楊輝三角形:第k+1行的第i個元素,等於第k個元素的第i個元素加第i-1個元素

2)爲滿足空間複雜度爲O(k)的需求,可以從第2行開始,每次在list末尾添加一個1,則前k個元素爲k-1行的值,這樣就可以從後向前計算第k行的元素


AC代碼

/* 
        1 
       1 1 
      1 2 1 
     1 3 3 1 
    1 4 6 4 1 
*/  
public class Solution {
    public List<Integer> getRow(int rowIndex) {
        List<Integer> list = new ArrayList<Integer>(rowIndex + 1);
        
        list.add(1);
        if(rowIndex == 0){
            return list;
        }
        
        list.add(1);
        if(rowIndex == 1){
            return list;
        }
        
        for(int i=2; i<=rowIndex; i++){
            list.add(1);
            for(int j=i-1; j>0; j--){
                list.set(j, list.get(j) + list.get(j-1));
            }
        }
        return list;
    }
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章