[Leetcode] 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?

分析

给定 n 返回杨辉三角的第 n 行。

仿照 前一题 [Pascal’s Triangle] 的思路进行构造。由于这里不需要返回前 n 行而只需要第 n 行，因此可以共用一个向量，每次将杨辉三角的下一行赋值给这个向量，直到第 n 行，这样空间复杂度为 0(n) 。

代码

class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> res(rowIndex + 1, 0);
for (int i = 0; i <= rowIndex; i++) {
res[0] = res[i] = 1;
for (int j = i - 1; j > 0; j--)
res[j] += res[j - 1];
}
return res;
}
};