题目描述
给定一个非负索引 k,其中 k ≤ 33,返回杨辉三角的第 k 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。
示例:
输入: 3
输出: [1,3,3,1]
进阶:
你可以优化你的算法到 O(k) 空间复杂度吗?
问题分析
要达到题目要求的空间复杂度,全算是不可能的。看了几个题解,都是在一行中覆盖值以达到节省空间的目的。
代码
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> res(rowIndex + 1);
res[0] = 1;
for (int i = 1; i <= rowIndex; ++i) {
for (int j = i; j >= 1; --j) {
res[j] += res[j - 1];
}
}
return res;
}
};
参考
https://www.cnblogs.com/grandyang/p/4031536.html
还有一个柳神的解法,我没有看懂,看起来是用了一个规律。在此仅做记录。
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> res(rowIndex + 1);
res[0] = 1;
for (int i = 1; i <= rowIndex; i++) {
res[i] = (long long int)res[i - 1] *(long long int)(rowIndex - i + 1) / i;
}
return res;
}
};