【Question】
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?
方法:题目要求返回第kth行的序列,这个不难。但是题目还要求额外的空间为O(k) 。所以新建序列来做的序列行不通。
然后杨辉三角是对称的,我们只需修改一般的序列就可以了。
对于K=3的情况,首先需要得到k=2【1 2 1】,由于下一行比上一行多一个元素
首先,在序列的末尾添加元素1,因为杨辉三角的末尾都是1,得到【1 2 1 1】;
然后,先保持序列的前1/2元素的值不变,修改后半序列的值;
接着,从倒数第二个元素开始修改,根据对称性倒数第二个元素与正数第二个元素的只应该相等,倒数第二个元素的值=正数第二个值=上一行第一个值(也就是此时的第一个元素)+上一行第二个值(也就是此时的第二个元素);
最后,修改完后半部分的序列后,根据对称,将前半部分的值进行修改;
依次循环下去,就会得到第N行的序列
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> tria(NULL);
if (rowIndex<0) return tria;
tria.push_back(1);
if(rowIndex==0) return tria;
tria.push_back(1);
if(rowIndex==1) return tria;
int i=1;
while(i!=rowIndex)
{
tria.push_back(1);
int len =tria.size();
int j;
for(j=1;j<=(len-1)/2;j++)
tria[len-j-1]=tria[j-1]+tria[j];
for(j=1;j<=(len-1)/2;j++)
tria[j]=tria[len-j-1];
i++;
}
return tria;
}
};