[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's Triangle的升级版，当然思路大致是一样的，leetcode有很多这种升级版的题目，都可以用原来的版本改良得到。这题不要求列出从1—n所有行了，只要求拿出第k行。当然要得到第k行也要从第一行开始推进，只是这里不再需要保存k－1以前的结果，因为k的结果只跟k－1有关，所以只需要O(k)的额外空间。 这里要注意一点的是k是从0开始的，而上一题是从1开始的＝_=

void getRowIter(int index, int n, vector<int> &aux) {
    if (index > n) return;
    vector<int> t;
    
    t.push_back(1);
    for (int k = 0; k < index-1; k++) {
        t.push_back(aux[k]+aux[k+1]);
    }
    t.push_back(1);
    aux = t;
    getRowIter(index+1, n, aux);
}


vector<int> getRow(int rowIndex) {
    vector<int> aux;
    if (rowIndex < 0) return aux;
    aux = {1};
    if (rowIndex == 0) return aux;
    getRowIter(1,rowIndex, aux);
    return aux;
}