首页 > 代码库 > [LeetCode] Pascal's Triangle II

[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;}

 

[LeetCode] Pascal's Triangle II