链接
https://leetcode-cn.com/problems/pascals-triangle-ii/
耗时
解题:18 min
题解:18 min
题意
给定一个非负索引 k,其中 k ≤ 33,返回杨辉三角的第 k 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。
思路
将杨辉三角形放在二维数组的左下三角形,根据定义可以得到每个位置的元素值的递推式为:
d p [ i ] [ j ] = d p [ i − 1 ] [ j − 1 ] + d p [ i − 1 ] [ j ] dp[i][j] = dp[i-1][j-1] + dp[i-1][j] dp[i][j]=dp[i−1][j−1]+dp[i−1][j]
滚动数组优化空间复杂度:
d p [ j ] = d p [ j − 1 ] + d p [ j ] dp[j] = dp[j-1] + dp[j] dp[j]=dp[j−1]+dp[j]
时间复杂度: O ( k 2 ) O(k^2) O(k2)
AC代码
class Solution {
public:
vector<int> getRow(int rowIndex) {
vector<int> dp(rowIndex+1, 0);
dp[0] = 1;
for(int i = 1; i <= rowIndex; ++i) {
for(int j = i; j > 0; --j) {
dp[j] += dp[j-1];
}
}
return dp;
}
};