LeetCode 119. 杨辉三角 II

SakuraTears

发布于 2022-01-13 14:30:21

1300

发布于 2022-01-13 14:30:21

文章被收录于专栏：从零开始的Code生活

题目

给定一个非负索引 k，其中 k ≤ 33，返回杨辉三角的第 k 行。在杨辉三角中，每个数是它左上方和右上方的数的和。

示例:

输入: 3 输出: [1,3,3,1] 进阶：

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

思路

将杨辉三角存到一个二维数组中打表，然后在求所需索引

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

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2021年02月12日，如有侵权请联系 cloudcommunity@tencent.com 删除

数组