[Leetcode][JAVA] Pascal's Triangle I, II

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]
]


已知行数生成帕斯卡三角。实际上只要有第i层，那么就能生成第i+1层。每次新生成的层加入最终集合中即可。

    public List<List<Integer>> generate(int numRows) {
        List<List<Integer>> re = new ArrayList<List<Integer>>();
        
        for(int i=0;i<numRows;i++) {
            List<Integer> temp = new ArrayList<Integer>();
            for(int j=0;j<=i;j++) {
                if(j==0 || j==i)
                    temp.add(1);
                else
                    temp.add(re.get(i-1).get(j-1)+re.get(i-1).get(j));
            }
            re.add(temp);
        }
        return re;
    }

 

Pascal's Triangle II：

Given an index k, return the kth row of the Pascal's triangle.

For example, given k = 3,
Return [1,3,3,1].

Note:
Could you optimize your algorithm to use only O(k) extra space?

 与第一题几乎一样，只不过不需要返回整个三角而只需要返回最后一层。全程只需要维护生成层和它上一层两个ArrayList即可。

    public List<Integer> getRow(int rowIndex) {
        List<Integer> re = new ArrayList<Integer>();
        for(int i=0;i<=rowIndex;i++) {
            List<Integer> temp = new ArrayList<Integer>();
            for(int j=0;j<=i;j++) {
                if(j==0 || j==i)
                    temp.add(1);
                else
                    temp.add(re.get(j-1) + re.get(j));
            }
            re = temp;
        }
        return re;
    }