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