【LeetCode】118_Pascal's Triangle

题目

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

解析

很简单，下一行第j元素用上一行相同的位置和前一个位置的和即可。需要注意的就是两边都是1

直接上代码

class Solution {
public:
    vector<vector<int>> generate(int numRows){
		vector<vector<int>> ret;
		for (int i = 0;i<numRows;i++)
		{
			vector<int> tmp(i+1);
			tmp[0] = 1;
			tmp[i] = 1;
			for (int j = 1;j<i;j++)
			{
				tmp[j] = ret[i-1][j]+ret[i-1][j-1];
			}
			ret.push_back(tmp);
		}
		return ret;
	}
};

方法挺简单的，可以优化的地方在于其实他是左右对称的，没必要把一行全计算一遍。

看看大神的代码

从左到右

class Solution {
public:
    vector<vector<int>> generate(int numRows){
        vector<vector<int> > result;
        if(numRows == 0) return result;
        result.push_back(vector<int>(1,1)); //first row
        for(int i = 2; i <= numRows; ++i) 
        {
            vector<int> current(i,1); // 本行
            const vector<int> &prev = result[i-2]; // 上一行
            for(int j = 1; j < i - 1; ++j) 
            {
                current[j] = prev[j-1] + prev[j]; // 左上角和右上角之和
            }
            result.push_back(current);
        }
    return result;
	}
};

从右到左

class Solution {
public:
    vector<vector<int>> generate(int numRows){
       vector<vector<int> > result;
       vector<int> array;
        for (int i = 1; i <= numRows; i++) 
        {
            for (int j = i - 2; j > 0; j--) 
            {
                array[j] = array[j - 1] + array[j];
            }
            array.push_back(1);
            result.push_back(array);
        }
        return result;
	}
};