给定一个非负索引 k,其中 k ≤ 33,返回杨辉三角的第 k 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。
示例:
输入: 3
输出: [1,3,3,1]
进阶:
你可以优化你的算法到 O(k) 空间复杂度吗?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/pascals-triangle-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路:
优化算法到O(K)空间复杂度,关键在于用好杨辉三角的公式,即前者与后者的比与公式有关,步骤如下:
- 获取杨辉三角的指定行
- 直接使用组合公式C(n,i) = n!/(i!*(n-i)!)
- 则第(i+1)项是第i项的倍数=(n-i)/(i+1);
代码如下:
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> ans;
long cur = 1;
for(int i = 0; i <= rowIndex; i ++) {
ans.push_back((int)cur);
// 注意这里有运算符先后问题,要先乘再除!
cur = cur * (rowIndex - i) / (i + 1);
}
return ans;
}
};