LeetCode *** 119. Pascal's Triangle II

最新推荐文章于 2024-05-25 20:57:01 发布

原创最新推荐文章于 2024-05-25 20:57:01 发布 · 200 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

212 篇文章

订阅专栏

本文介绍了一种高效求解杨辉三角指定行的算法，并将其空间复杂度优化到O(k)，通过逐行计算并复用数组实现。文章提供了完整的C++代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：

Given an index k, return the k^th row of the Pascal's triangle.

For example, given k = 3,
Return [1,3,3,1].

Note:
Could you optimize your algorithm to use only O(k) extra space?

分析：

杨辉三角。左右相加。没啥说的。

代码：

最初：

class Solution {
public:
    vector<int> getRow(int rowIndex) {
        
        vector<int> pre;
        
        for(int i=1;i<=rowIndex+1;++i){
            
            vector<int> res;
            
            res.insert(res.begin(),1);
            if(i!=1)res.insert(res.end(),1);
            
            for(int j=1;j<i-1;++j){
                res.insert(res.end()-1,pre[j-1]+pre[j]);
            }
            
            pre=res;
            
        }
        
        return pre;
        
    }
};

最终：

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>0;--j)
                res[j]+=res[j-1];
                
        return res;
        
    }
};