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层。每次新生成的层加入最终集合中即可。
1 public List<List<Integer>> generate(int numRows) { 2 List<List<Integer>> re = new ArrayList<List<Integer>>(); 3 4 for(int i=0;i<numRows;i++) { 5 List<Integer> temp = new ArrayList<Integer>(); 6 for(int j=0;j<=i;j++) { 7 if(j==0 || j==i) 8 temp.add(1); 9 else 10 temp.add(re.get(i-1).get(j-1)+re.get(i-1).get(j)); 11 } 12 re.add(temp); 13 } 14 return re; 15 }
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即可。
1 public List<Integer> getRow(int rowIndex) { 2 List<Integer> re = new ArrayList<Integer>(); 3 for(int i=0;i<=rowIndex;i++) { 4 List<Integer> temp = new ArrayList<Integer>(); 5 for(int j=0;j<=i;j++) { 6 if(j==0 || j==i) 7 temp.add(1); 8 else 9 temp.add(re.get(j-1) + re.get(j)); 10 } 11 re = temp; 12 } 13 return re; 14 }