【Leetcode】Pascal's Triangle II

最新推荐文章于 2019-06-24 19:31:47 发布

原创最新推荐文章于 2019-06-24 19:31:47 发布 · 1.2k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #leetcode #pascals triangle #滚动数组

leetcode题目摘选专栏收录该内容

38 篇文章

订阅专栏

本文介绍了一种高效算法来求解帕斯卡三角形的第K行，使用滚动数组实现仅需O(K)的额外空间。通过具体代码示例展示了如何利用这种方法进行计算。

题目：

Given an index k, return the kth 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> result;
        if(rowIndex<0)return result;
        result.push_back(1);
        for(int i=0;i<rowIndex;i++){
            for(int j=i;j>0;j--){
                result[j]=result[j]+result[j-1];
            }
            result.push_back(1);
        }
        return result;
    }
};