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?
下面程序空间效率O(k),使用了两个vector<int>交替使用,因为只利用上一行就能填写下一行了。
vector<int> getRow(int rowIndex)
{
vector<int> pre;
if (rowIndex < 0) return pre;
vector<int> rs(1,1);
for (int i = 0; i < rowIndex; i++)
{
pre.clear();
rs.swap(pre);
rs.push_back(1);
for (int j = 0; j < i; j++)
{
rs.push_back(pre[j]+pre[j+1]);
}
rs.push_back(1);
}
return rs;
}
因为只利用当前行两个数值的信息覆盖新填写的数列格也是可以的,所以只利用一个vector<int>也是可以的。
Leetcode上总有人写出更加简洁的程序的:
vector<int> getRow(int rowIndex)
{
vector<int> array;
for (int i = 0; i <= rowIndex; i++)
{
for (int j = i-1; j > 0; j--)
{
array[j] = array[j-1] + array[j];
}
array.push_back(1);
}
return array;
}
//2014-2-17 update
vector<int> getRow(int rowIndex)
{
vector<int> rs(1,1);//注意题意:当rowIndex==0是output:[[1]]
for (int i = 1; i <= rowIndex; i++)
{
for (int j = i-1; j > 0; j--)
{
rs[j] += rs[j-1];//注意:这里需要逆向填表!
}
rs.push_back(1);
}
return rs;
}