leetcode.118 楊輝三角
阿新 • • 發佈:2018-11-14
給定一個非負整數 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; } }