Given an index k, return the kth row of the Pascal's triangle.
For example, given k = 3,
Return [1,3,3,1]
.
方法一:
由于之前我们曾经遇到过建立Pascal三角形,所以我们自然联想到建立一个K+1的三角形,然后取出来第K行,就得到所求的结果。
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<vector<int>> temp;
for(int i=0;i<rowIndex+1;i++)
{
vector<int> tem;
for(int j=0;j<=i;j++)
{
if(j==0||j==i)
tem.push_back(1);
else{
tem.push_back(temp[i-1][j-1]+temp[i-1][j]);
}
}
temp.push_back(tem);
}
return temp[rowIndex];
}
};
方法二:
可以维护一个一维数组,并且不断地修改它,使得它的成为第k行的Pascal's Triangle.
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> res;
if(rowIndex<0)
return res;
res.push_back(1);
for(int i=1;i<rowIndex+1;i++)
{
int pre=0;
for(int j=0;j<res.size();j++)
{
int cur=res[j];
res[j]=pre+res[j];
pre=cur;
}
res.push_back(1);
}
return res;
}
};