[leetcode] Pascal's Triangle
Given numRows, generate the first numRows of Pascal's triangle.
For example, given numRows = 5,
Return
[ [1], [1,1], [1,2,1], [1,3,3,1], [1,4,6,4,1] ]
https://oj.leetcode.com/problems/pascals-triangle/
思路:按照规律根据上一行的数据生成即可。
public class Solution { public List<List<Integer>> generate(int numRows) { List<List<Integer>> res = new ArrayList<List<Integer>>(); if(numRows<=0) return res; List<Integer> last = new ArrayList<Integer>(); for(int i=0;i<numRows;i++){ List<Integer> cur = new ArrayList<Integer>(); for(int j=0;j<=i;j++){ if(j==0||j==i) cur.add(1); else cur.add(last.get(j-1)+last.get(j)); } res.add(cur); last=cur; } return res; } }
第二遍记录:注意cur每次都生成新的,循环完一行后赋值给pre。
public class Solution { public List<List<Integer>> generate(int numRows) { List<List<Integer>> res = new ArrayList<List<Integer>>(); if(numRows == 0) return res; List<Integer> pre = new ArrayList<Integer>(); for(int i=0;i < numRows;i++){ List<Integer> cur = new ArrayList<Integer>(); for(int j=0;j<=i;j++){ if(j==0||j==i){ cur.add(1); }else{ cur.add(pre.get(j-1)+pre.get(j)); } } res.add(cur); pre =cur; } return res; } }