方法一:
如果解了杨辉三角I,其实可以偷懒一下,就把其中要求的那一行返回就行了。但是这样比较浪费空间。
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<vector<int>>result(rowIndex+1);
result[0].push_back(1);
if(rowIndex==0)return result[0];
for(int i=1;i<=rowIndex;i++){
for(int j=0;j<=i;j++){
if(j==i||j==0)result[i].push_back(1);
else result[i].push_back(result[i-1][j-1]+result[i-1][j]);
}
}
return result[rowIndex];
}
};
方法二:
因为只需要一行,所以就不用存整个二维数组,只需要对一行数组进行不断的变化就可以。可以设置一个temp来存储上一行的数组。但是也可以不设temp
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int>res(rowIndex+1,0);
res[0]=1;
for(int i=1;i<rowIndex+1;i++){
for(int j=i;j>=1;j--){
res[j]=res[j]+res[j-1];
}
}
return res;
}
};
这样每次都替换它自己本身的位置
方法三:
借鉴LeetCode上大佬题解
杨辉三角是有推导公式的:
而且前后组合数是有联系的,可以推出:
这样就只需要一个for循环就可以解决,复杂度O(K)
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int>res;
long pre=1,cur=1;//注意层数多了会超出int最大值的范围,使用long可以表示更大的数字
res.push_back(1);
for(int k=1;k<=rowIndex;k++){
pre=cur*(rowIndex-k+1)/k;
res.push_back(pre);
cur=pre;
}
return res;
}
};