118. 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]
]
其实每一列都是数学里的排列组合, 第4行, 可以用 C30 = 0 C31=3 C32=3 C33=3 来求。(ps:如果行数大于等于30,由于temp声明为int,
会超过范围,可把代码中的integer换成Long,int换成long。。。。所以更建议用第二种方法。)
public class PascalTriangle1 { public static void main(String[] args) { System.out.println(generate(5)); } public static List<List<Integer>> generate(int numRows) { List<List<Integer>> list = new ArrayList<List<Integer>>(numRows); for (int i = 0; i < numRows; i++) { List<Integer> thisRow = new ArrayList<Integer>(i); thisRow.add(1); int temp = 1; int row = i; for (int j = 1; j < i+1; j++) { temp = temp * row-- /j; thisRow.add(temp); } list.add(thisRow); } return list; } }
或者由下一层的元素是由上一层的元素生成的可以换一种算法,顺便把杨辉三角的第二种问题用这个方法解决
Given an index k, return the kth row of the Pascal's triangle.
For example, given k = 3,
Return[1,3,3,1]
.
public static List<Integer> generate2(int rowIndex) { int numRows =rowIndex+1; List<List<Integer>> result = new ArrayList<List<Integer>>(); //第一行 ArrayList<Integer> first = new ArrayList<Integer>(); first.add(1); result.add(first); for(int n = 2; n <= numRows; n++){ ArrayList<Integer> thisRow = new ArrayList<Integer>(); //每一行的第一个元素都是1 thisRow.add(1); //中间元素由上一行元素得出 //A(n+1)[i] = A(n)[i - 1] + A(n)[i] List<Integer> previousRow = result.get(n- 2); for(int i = 1; i < n - 1; i++){ thisRow.add(previousRow.get(i - 1) + previousRow.get(i)); } //每一行的最后一个元素都是1 thisRow.add(1); result.add(thisRow); } return result.get(rowIndex); }