【Pascal's Triangle II 】cpp

题目:

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> ret(rowIndex+1,1);
            for ( int i=0; i<=rowIndex; ++i )
            {
                for (int j=i-1; j>0; --j)
                {
                    ret[j] = ret[j] + ret[j-1];
                }
            }
            return ret;
    }
};

tips:

采用滚动数组技巧,可以缩减空间复杂度。

==========================================

第二次过这道题,题意一开始没有看清,改了一次AC了。

class Solution {
public:
    vector<int> getRow(int rowIndex) {
            vector<int> ret(rowIndex<1?1:rowIndex+1,0);
            ret[0] = 1;
            for ( int i=1; i<=rowIndex; ++i )
            {
                for ( int j=i; j>0; --j )
                {
                    ret[j] = ret[j-1] + ret[j];
                }
            }
            return ret;
    }
};

 

posted on 2015-06-08 23:28  承续缘  阅读(155)  评论(0编辑  收藏  举报

导航