题目:
给定一个非负索引 k,其中 k ≤ 33,返回杨辉三角的第 k 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。
示例:
输入: 3
输出: [1,3,3,1]
思路:从第一行开始,按照规律计算,每次下一行的结果覆盖上一行的结果,直到指定的那一行为止。
代码实现:
class Solution {
public:
vector<int> getRow(int rowIndex) {
int num1,num2,k=0;
vector<int> v1;
vector<int> v2;
v1.push_back(1);
for(int i=0;i<=rowIndex;i++){
//每个数是它左上方和右上方的数的和
while(k+1 <= v1.size()-1){
num1 = v1[k];
num2 = v1[k+1];
v2.push_back(num1+num2);
k++;
}
//最后一个元素是1
v2.push_back(1);
k=0;
v1.clear();
//到达指定行结束
if(i == rowIndex){
break;
}
v1 = v2;
v2.clear();
//第1个元素是1
v2.push_back(1);
}
return v2;
}
};
效率:
简洁版:
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> result;
for(int i = 0; i <= rowIndex; ++i){
result.push_back(1);
for(int j = i - 1; j > 0; --j){
result[j] += result[j - 1];
}
}
return result;
}
};
效率:
进阶:
你可以优化你的算法到 O(k) 空间复杂度吗?
参考解法:
/**
* 获取杨辉三角的指定行
* 直接使用组合公式C(n,i) = n!/(i!*(n-i)!)
* 则第(i+1)项是第i项的倍数=(n-i)/(i+1);
*/
class Solution {
public List<Integer> getRow(int rowIndex) {
List<Integer> res = new ArrayList<>(rowIndex + 1);
long cur = 1;
for (int i = 0; i <= rowIndex; i++) {
res.add((int) cur);
cur = cur * (rowIndex-i)/(i+1);
}
return res;
}
}
效率: