码迷,mamicode.com
首页 > 编程语言 > 详细

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

时间:2015-01-08 07:02:25      阅读:225      评论:0      收藏:0      [点我收藏+]

标签:

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     }

 

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

标签:

原文地址:http://www.cnblogs.com/splash/p/4209855.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!