杨辉三角形II

余生长醉 提交于 2020-02-07 06:04:29

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

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

示例:

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

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/pascals-triangle-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

class Solution {
public:
    vector<int> getRow(int rowIndex) {
        //这道题是在杨辉三角形上的晋级;
        //先按照杨辉三角形的方法接一下;
        vector<vector<int>>res(rowIndex+1);
        if(rowIndex==0){
            return {1};
        }
        res[0].push_back(1);
        for(int i=1;i<rowIndex+1;i++){
            res[i].push_back(1);
            for(int j=1;j<i;j++){
                res[i].push_back(res[i-1][j-1]+res[i-1][j]);
            }
            res[i].push_back(1);
        }
        return res[rowIndex];
    }
};

解法二:
只用有限的空间复杂度解决:

class Solution {
public:
    vector<int> getRow(int rowIndex) {
        //这道题是在杨辉三角形上的晋级;
        //空间复杂度是k
        if(rowIndex==0){
            return {1};
        }
        vector<int>res;
        res.push_back(1);
        for(int i=0;i<rowIndex;i++){
            res.insert(res.begin(),0);
            for(int j=0;j<res.size()-1;j++){
                res[j] = res[j]+res[j+1];
            }
            
        }
        return res;
    }
};
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!