LeetCode 119. 杨辉三角 II

最新推荐文章于 2024-07-27 19:43:02 发布

原创最新推荐文章于 2024-07-27 19:43:02 发布 · 146 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

261 篇文章

订阅专栏

本文深入解析了杨辉三角的算法优化，针对第k行的求解，提出了一种O(k)的空间复杂度解决方案。通过使用临时数组存储前一行的数据，实现了在计算当前行时对所需历史数据的复用，从而避免了全三角形的存储，显著降低了空间消耗。

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

题目描述

给定一个非负索引 k，其中 k ≤ 33，返回杨辉三角的第 k 行。

在杨辉三角中，每个数是它左上方和右上方的数的和。

示例:

输入: 3
输出: [1,3,3,1]

进阶：

你可以优化你的算法到 O(k) 空间复杂度吗？

问题分析

此题和上一题思想是一样的，只不过我们用一个temp数组，分别在本行元素都创建出来并保存到out后，将其也保存到temp中，这样接下来后面在进行下一行的创建时，会用到当前行的元素信息，就从temp数组中获取。每一行在元素创建出来之后，判断如果当前行就是我们最后要输出的行，那么我们将out赋给ans。那么循环接下来会退出然后将ans返回。

代码实现

class Solution {
public:
    vector<int> getRow(int rowIndex) {
        vector<int> temp;
        vector<int> ans;
        for(int i = 0; i < rowIndex + 1; i++){
            vector<int> out;
            if(i == 0){
                out.push_back(1);
                if(i == rowIndex)
                    ans = out;
                temp = out;
            }
            if(i == 1){
                out.push_back(1);
                out.push_back(1);
                if(i == rowIndex)
                    ans = out;
                temp = out;
            }
            if(i > 1){
                out.push_back(1);
                for(int j = 1; j < i; j++)
                    out.push_back(temp[j-1] + temp[j]);
                out.push_back(1);
                if(i == rowIndex)
                    ans = out;
                temp = out;
            }
            temp.clear();
        }
        return ans;
    }
};