[Leetcode从零开刷]119. Pascal's Triangle II

题目来源

leetcode


题目要求:

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?

题目解答:

这个和(18. Pascal’s Triangle)题目一样,最傻逼的办法就是用这个代码,返回最后一个就行,但是空间超了,也可以accepted·

class Solution {
    public List getRow(int rowIndex) {
         List> res = new ArrayList<>();
        List row,pre = null;
        for(int i=0;i<=rowIndex;i++){
            row = new ArrayList<>();
            for(int j=0;j<=i;j++){
                if(j==0||j==i)
                    row.add(1);
                else
                    row.add(pre.get(j-1)+pre.get(j));
            }
            pre = row;
            res.add(row);
        }
        return res.get(res.size()-1);     
    }
}

厉害的c++

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

你可能感兴趣的:(Algorithm,problem,刷题-从零开刷)