简单难度
题目描述
给定一个非负索引 rowIndex
,返回「杨辉三角」的第 rowIndex
行。
在「杨辉三角」中,每个数是它左上方和右上方的数的和。
示例 1:
输入: rowIndex = 3
输出: [1,3,3,1]
示例 2:
输入: rowIndex = 0
输出: [1]
示例 3:
输入: rowIndex = 1
输出: [1,1]
提示:
0 <= rowIndex <= 33
进阶:
你可以优化你的算法到 O(rowIndex)
空间复杂度吗?
结果
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<vector<int> > tmp;
vector<int> res;
for (int i = 0; i <= rowIndex; ++i) {
res.clear();
for (int j = 0; j <= i; ++j) {
if (0 == j || i == j) res.emplace_back(1);
else res.emplace_back(tmp[i - 1][j] + tmp[i - 1][j - 1]);
}
tmp.emplace_back(res);
}
return res;
}
};
优化
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> row(rowIndex + 1);
row[0] = 1;
for (int i = 1; i <= rowIndex; ++i)
row[i] = 1LL * row[i - 1] * (rowIndex - i + 1) / i;
return row;
}
};