leetcode 119. 杨辉三角 II
问题描述
给定一个非负索引 k,其中 k ≤ 33,返回杨辉三角的第 k 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。
示例:
输入: 3
输出: [1,3,3,1]
进阶:
你可以优化你的算法到 O(k) 空间复杂度吗?
问题分析
118. 杨辉三角要输出整个三角,因此空间复杂度为\(O(N^2)\).从下图来看,每一行的数据我们可以倒着向前计算,从而避免了覆盖数据的问题:
1
1 1
1 2 1
1 3 3 1
1 4 6 4 1
代码
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> ans(rowIndex+1,1);
for(int i = 0; i < (rowIndex+1); i++)
{
ans[i] = 1;
for(int j = i-1; j > 0; --j)
{
ans[j] += ans[j-1];
}
}
return ans;
}
};
结果:
执行用时 :0 ms, 在所有 C++ 提交中击败了100.00%的用户
内存消耗 :8.1 MB, 在所有 C++ 提交中击败了99.86%的用户