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]
]
遍历修改,一开始的代码如下:
public class Solution {
public List<List<Integer>> generate(int numRows) {
List<List<Integer>> pascal = new ArrayList<List<Integer>> ();
for(int i = 0; i < numRows; i++) {
List<Integer> list = new ArrayList<Integer> ();
list.add(1);
if(i > 1) {
for(int j = 1; j < i; j++) {
list.add(j, pascal.get(i - 1).get(j) + pascal.get(i - 1).get(j - 1));
}
}
if(i > 0) list.add(1);
pascal.add(list);
}
return pascal;
}
}
考虑到每次新的 list 可以在上一个 list 上直接修改,这样可以减少计算步骤,代码如下:
public class Solution {
public List<List<Integer>> generate(int numRows) {
List<List<Integer>> pascal = new ArrayList<List<Integer>> ();
List<Integer> list = new ArrayList<Integer> ();
for(int i = 0; i < numRows; i++) {
for(int j = i - 1; j > 0; j--) {
list.set(j, list.get(j) + list.get(j - 1));
}
list.add(1);
pascal.add(new ArrayList<Integer>(list));
}
return pascal;
}
}