LeetCode 119 杨辉三角II HERODING的LeetCode之路

给定一个非负索引 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;
    }
};

猜你喜欢

转载自blog.csdn.net/HERODING23/article/details/113792839