码迷,mamicode.com
首页 > 其他好文 > 详细

LeetCode118 Pascal's Triangle

时间:2016-11-22 22:54:19      阅读:192      评论:0      收藏:0      [点我收藏+]

标签:new   else   turn   pascal   代码   三角形   二维数组   杨辉三角   class   

Given numRows, generate the first numRows of Pascal‘s triangle.

For example, given numRows = 5, (Easy)
Return

[
     [1],
    [1,1],
   [1,2,1],
  [1,3,3,1],
 [1,4,6,4,1]
]

 

分析:

求解杨辉三角形,按照其定义以此求解即可,注意优化写法使其能够更简洁(比如对于二维数组每一行的个数其实是已知的,所以可以先push_back一个长度的数组,后面直接用下标访问),即第二种写法的代码。

代码:

 1 class Solution {
 2 public:
 3     vector<vector<int>> generate(int numRows) {
 4         vector<vector<int>> result;
 5         if (numRows == 0) {
 6             return result;
 7         }
 8         vector<int> oldVec{1};
 9         result.push_back(oldVec);
10         for (int i = 2; i <= numRows; ++i) {
11             vector<int> newVec;
12             for (int j = 0; j < i; ++j) {
13                 int temp;
14                 if (j > 0 && j < oldVec.size()) {
15                     temp = oldVec[j] + oldVec[j - 1];
16                 }
17                 else if (j > 0){
18                     temp = oldVec[j - 1];
19                 }
20                 else {
21                     temp = oldVec[0];
22                 }
23                 newVec.push_back(temp);
24             }
25             result.push_back(newVec);
26             oldVec = newVec;
27         }
28         return result;
29     }
30 };
 1 class Solution {
 2 public:
 3     vector<vector<int>> generate(int numRows) {
 4         vector<vector<int>> result;
 5         if (numRows == 0) {
 6             return result;
 7         }
 8         vector<int> init{1};
 9         result.push_back(init);
10         for (int i = 2; i <= numRows; ++i) {
11             result.push_back(vector<int>(i,1));
12             for (int j = 1; j < i - 1; ++j) {
13                 result[i - 1][j] = result[i - 2][j] + result[i - 2][j - 1];
14             }
15         }
16         return result;
17     }
18 };

 

LeetCode118 Pascal's Triangle

标签:new   else   turn   pascal   代码   三角形   二维数组   杨辉三角   class   

原文地址:http://www.cnblogs.com/wangxiaobao/p/6091148.html

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