【LeetCode刷题记录】Pascal's Triangle II

题目

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?

解答

这题和Pascal Triangle相近，要求返回第k行的数（注意：k从0开始，最开始被坑了T_T）。题目要求是空间复杂度为O(k)，也就是说申请的数组或vector最多只能有k个元素。本着“先解决问题”的原则，第一次暂时不考虑这个条件，照搬Pascal Triangle的计算公式，利用递归（效率不高）的思路实现。思路很简单，直接上AC代码：

class Solution {
public:
    vector<int> getRow(int rowIndex) {
        vector<int> row;// the kth row
        row.resize(rowIndex+1);

        if (rowIndex == 0) {// the 1st row
            row[0] = 1;
        }
        else if (rowIndex == 1) {// the 2nd row
            row[0] = row[1] = 1;
        }
        else {// the other row
            vector<int> rowPre;
            rowPre.resize(rowIndex);
            rowPre = getRow(rowIndex-1);
            row[0] = row[row.size()-1] = 1;//take care of the begin and end

            for(int i=1; i<row.size()-1; i++) {
                row[i] = rowPre[i-1] + rowPre[i];
            }
        }

        return row;
    }
};

然后，再考虑空间复杂度的问题。自己没想出来，去评论区里看牛人们的解法，一开始也不知所云。只好采用带入具体情况去推理的笨办法，总算是把思路理清楚了，分析如下：

本质上，求出第k行的值需要依赖第(k-1)行的值。在上面的笨方法中，另外开辟了空间去保存第(k-1)行的值。其实，一旦确定要行号k，那么第k行的大小就确定为k+1。申请一个k+1大小的vector后，从第0行开始，一行一行地计算，前面行得值直接存储在最后要返回的k+1行的vector中。
要注意的是，由于如果沿袭”row[i]=row[i-1]+row[i]”的公式，从前往后去计算，那么在计算第i个元素值后，row[i]会被新值覆盖，但是计算第i+1个元素值时，又需要原来的row[i]。比如，要求第3行，现在求出了第2行的结果row[4]={1,2,1,0}。从前往后计算的话，row[4]={1,3,4,4}；从后往前计算的话，row[4]={1,3,3,1}。因此，考虑从后往前计算，那么就不会存在覆盖的问题了。

分析略显啰嗦，主要是为了自己能看懂(^__^) 。第二次AC代码如下：

class Solution {
public:
    vector<int> getRow(int rowIndex) {
        vector<int> row;
        row.resize(rowIndex+1);// the kth row has (k+1) elements

        row[0] = 1;
        for (int i=1; i<=rowIndex; i++) {// from 1st to kth row
            for (int j=i; j>=1; j--) {// from row[k] to row[1]
                row[j] += row[j-1];
            }
        }

        return row;
    }
};

---

小结：后面刷leetcode的原则还是“先解决，再优化”，目前coding水平不高，只有务实基础才是相对适合自己的方法。欢迎各位大牛拍砖指正，多多赐教！