1. 程式人生 > >leetcode.118 楊輝三角

leetcode.118 楊輝三角

給定一個非負整數 numRows,生成楊輝三角的前 numRows 行。

在楊輝三角中,每個數是它左上方和右上方的數的和。

示例:

輸入: 5
輸出:
[
     [1],
    [1,1],
   [1,2,1],
  [1,3,3,1],
 [1,4,6,4,1]
]

思路:當 numRows 為 1 或者 2 的時候,值是固定的,分別是[ [1] ],[ [ 1 ] ,[ 1 ,1 ]],而從第三行中間的數是第二行的兩數之和

有 ,使用一個數組儲存上一行的值,並且使用另外的陣列記錄下要儲存的值。

程式碼:

class Solution {
    public List<List<Integer>> generate(int numRows) {
            if(numRows < 0) return null;
            List<List<Integer>> list = new ArrayList();
            if(numRows >= 1){
                List<Integer> data = new ArrayList();
                data.add(1);
                list.add(data);
        }
            if(numRows >= 2){
                List<Integer> data = new ArrayList();
                data.add(1);
                data.add(1);
                list.add(data);
        }
            if(numRows >= 3){
         
                for(int i = 3;i <= numRows;i++){
                    List<Integer> data = new ArrayList();
                    List<Integer> prev = list.get(i-2);
                    data.add(1);
                    for(int j = 2;j <= i-1 ;j++){
                           data.add(prev.get(j-2) + prev.get(i-1);
                    }
                    data.add(1);
                    list.add(data);
             }
         }
        return list;       
    }
}