[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<Integer> getRow(int rowIndex) {
         List<List<Integer>> res = new ArrayList<>();
        List<Integer> 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; i<rowIndex+1; i++)
            for(int j=i; j>=1; j--)
                A[j] += A[j-1];
        return A;
    }
};