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