题目描述
解法 模拟计算
根据杨辉三角的性质逐行计算
设一个开辟能容纳杨辉三角的二维数组,每一行通过变量i对应,列通过j控制:每行的元素对应dp[i][0] ~ dp[i][j]
容易发现除了边界为1,其他的公式为
dp[i][j] = dp[i - 1][j - 1] + dp[i - 1][j];
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<vector<int>> dp(rowIndex + 1, vector<int>(rowIndex + 1, 0));
dp[0][0] = 1;
for(int i = 1; i <= rowIndex; i++) {
for(int j = 0; j <= i; j++) {
if(j == 0 || j == i) {
dp[i][j] = 1;
} else {
dp[i][j] = dp[i - 1][j - 1] + dp[i - 1][j];
}
}
}
return dp[rowIndex];
}
};
空间复杂度O(k^2)
空间复杂度O(k^2)
优化
将空间复杂度降为O(k)
挺难想到的:
由于每一行在计算的时候只涉及 (上一行的 ( 对应当前元素下标的) 元素) 以及 (其上一个其元素)
所以只要在一个数组里面修改,就能达到换行的效果
同时为了保证上一个元素在本元素计算之前保留自己的值,应从尾到头反向遍历。
0号元素保持为1
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> res(rowIndex + 1, 0);
res[0] = 1;
for(int i = 1; i <= rowIndex; i++) {
for(int j = i; j > 0; j--) {
res[j] += res[j - 1];
}
}
return res;
}
};
时间复杂度O(k^2)
空间复杂度O(k)