题目:
给定一个非负索引 k,其中 k ≤ 33,返回杨辉三角的第 k 行。
要求:
O(k) 空间复杂度。
在杨辉三角中,每个数是它左上方和右上方的数的和。
示例:
输入: 3
输出: [1,3,3,1]
刚开始想按照上一题中生成杨辉三角的方式来做,最终输出某一行即可,但这样做的复杂度不是 O(k) 。进一步,只生成当前行和上一行,OJ还是通不过。
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> pre(rowIndex + 1, 1);
vector<int> cur(rowIndex + 1, 1);
int i, j;
for(i = 2; i < rowIndex; i++)
{
for(int j = 1; j < i; i++)
{
cur[j] = pre[j] + pre[j - 1];
}
pre = cur;
}
return cur;
}
};
继续优化。