力扣119题:从O(n²)到O(2n):杨辉三角高效空间优化

题目重解:
给定一个非负索引 rowIndex,返回杨辉三角的第 rowIndex 行。不同于生成整个杨辉三角,这道题要求我们只返回特定行,且空间复杂度应尽可能优化。例如输入3,需要返回[1,3,3,1]。
解题思路:
1.使用两个一维数组交替存储当前行和上一行数据
3.首尾元素固定为1,中间元素由上一行相邻元素相加得到
4.最终只需保留最后计算的行数据
代码详解:
class Solution {
public:
vector<int> getRow(int rowIndex) {
int a[2][34]; // 双数组存储空间
int now=1; // 当前写入数组索引
int pre=0; // 上一行数据数组索引
a[pre][0]=1; // 初始化第0行
// 逐行计算
for(int i=1;i<=rowIndex;i++) {
for(int j=0;j<=i;j++) {
if(j==i or j==0) { // 首尾元素为1
a[now][j]=1;
}
else { // 中间元素=上一行相邻元素之和
a[now][j]=a[pre][j]+a[pre][j-1];
}
}
now^=1; // 位运算切换数组
pre^=1; // 等价于now=(now+1)%2, pre=(pre+1)%2
}
// 组装结果
vector<int> v;
for(int i=0;i<=rowIndex;i++) {
v.push_back(a[pre][i]); // 注意最后使用pre指针
}
return v;
}
};原创内容 转载请注明出处





