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?
思路:
因为只返回结果一行,所以计算过程中除了k-1行有用之外,前面的空间都可以丢弃。
根据算法分析的规则,O(2k) = O(k)。因此最简单的方法就是用一个临时变量保存k-1行。然后不断更新这个变量。
注意交换时使用swap更新指针而不是vector赋值可以节约很多时间。
解:
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> resultRow = {1};
vector<int> tempRow;
for(int rowIdx = 0; rowIdx < rowIndex; rowIdx++)
{
tempRow.clear();
tempRow.push_back(1);
for(int idxInRow = 0; idxInRow < resultRow.size()-1; idxInRow++)
{
tempRow.push_back(resultRow[idxInRow]+resultRow[idxInRow+1]);
}
tempRow.push_back(1);
resultRow.swap(tempRow);
}
return resultRow;
}
};
直觉告诉我,这种递推的加和,一部分信息已经存在于遍历的次数中,所以应该有一种解法,可以直接操作resultVector,并节约50%的空间。
但因为数学天赋太差,我想不出来。
可以有时间时多想想。