LeetCode-----第119题-----杨辉三角 II

杨辉三角 II

难度简单163

给定一个非负索引 k，其中 k ≤ 33，返回杨辉三角的第 k 行。

在杨辉三角中，每个数是它左上方和右上方的数的和。

示例:

输入: 3
输出: [1,3,3,1]

进阶：

你可以优化你的算法到 O(k) 空间复杂度吗？

 

题目分析：

      在上一题的基础上进行修改，因为上一题记录了所有的层。我们知道计算当层的数只需要知道上一层即可，那么我们申请一个临时一维数组存放上一层的数，在计算新的结果即可，达到空间复杂度为O(k)

参考代码：

class Solution {
public:
    vector<int> getRow(int rowIndex) {
        vector<int> res;
		res.push_back(1);
		if (rowIndex < 1)
			return res;

		res.push_back(1);
		if (rowIndex == 1)
			return res;

		vector<int> res2;
		int j;
		for (int i = 2; i <= rowIndex; i++)
		{
			res2.push_back(1);
			for (j = 1; j < i; j++)
			{
				res2.push_back(res[j - 1] + res[j]);
			}
			res2.push_back(1);
            res = res2;
			res2.clear();
		}
		return res;
    }
};