118. Pascal's Triangle

来源：互联网发布：java ee 开源框架编辑：程序博客网时间：2024/05/20 16:11

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;    }}

0 0