leetcode_119. 杨辉三角 II

我们两清 提交于 2020-02-05 10:15:43

给定一个非负索引 k,其中 k ≤ 33,返回杨辉三角的第 k 行。
在这里插入图片描述
在杨辉三角中,每个数是它左上方和右上方的数的和。

示例:

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

进阶:

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

解析:本题与上一题的不同是仅需要输出一行,另外额外要求空间复杂度O(K)
因此我们仅设置一维数组,代表每一行的值,每次计算一行,然后用这一行的数值计算下一行并覆盖。

class Solution {
public:
    vector<int> getRow(int rowIndex) {
        vector<int> res(rowIndex+1);
        if(rowIndex==0){
            res[0]=1;
            return res;
        }
        else{
            vector<int> tmp = getRow(rowIndex-1);
            for(int i=0;i<rowIndex+1;++i){
                if(i==0||i==rowIndex) res[i]=1;
                else res[i]=tmp[i-1]+tmp[i];
            }
            return res;
        }
    }
};
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!