当前位置:首页 > 力扣 > 力扣119题:从O(n²)到O(2n):杨辉三角高效空间优化

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

2周前 (05-18)

力扣119题:从O(n²)到O(2n):杨辉三角高效空间优化 杨辉三角形 C++ 算法 力扣 滚动数组 数组 第1张


题目重解:

给定一个非负索引 rowIndex,返回杨辉三角的第 rowIndex 行。不同于生成整个杨辉三角,这道题要求我们只返回特定行,且空间复杂度应尽可能优化。例如输入3,需要返回[1,3,3,1]。


解题思路:

1.使用两个一维数组交替存储当前行和上一行数据

2.通过now/pre指针异或运算实现数组切换

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;
    }
};




原创内容 转载请注明出处

分享给朋友:

相关文章

力扣第75题新思路:如何用选择排序实现原地操作?

力扣第75题新思路:如何用选择排序实现原地操作?

给定一个包含红色、白色和蓝色元素的数组,分别用数字 0、1、2 表示,要求在不使用库排序函数的情况下,仅通过一次遍历(但实际上允许使用经典排序方法)对数组进行原地排序。题目要求将所有 0 排在前面,1...

力扣451:ASCII数组计数法 用128个桶解决频率排序问题

力扣451:ASCII数组计数法 用128个桶解决频率排序问题

题目重解给定一个字符串,将字符按照出现频率降序排列。例如输入"tree",可能返回"eetr"或"eert"。题目要求我们不考虑字母顺序,只...

力扣LCR182:字符串操作三连 从基础拼接到底层指针优化

力扣LCR182:字符串操作三连 从基础拼接到底层指针优化

题目重解需要将密码字符串从第target个字符开始进行重新排列,形成新的动态密码。例如输入"password"和target=3,结果应为"swordpas"。...

力扣面试题 16.01 :用异或操作玩转两数交换

力扣面试题 16.01 :用异或操作玩转两数交换

给定一个长度为 2 的整数数组 numbers,要求在不使用额外内存空间(即不使用临时变量)的情况下,交换数组中的两个元素并返回。题目考验对位运算的理解与应用,需通过巧妙的异或操作实现两数值...

力扣第92题:三步定位 精准反转链表指定区间

力扣第92题:三步定位 精准反转链表指定区间

题目解读给定一个单链表和两个整数left与right,要求将链表中从第left个节点到第right个节点的部分进行反转,而保持其他部分不变。例如,对于链表1→2→3→4→5,left=2,right=...

力扣1221:一次扫描解决分割平衡字符串 时间O(n)空间O(1)

力扣1221:一次扫描解决分割平衡字符串 时间O(n)空间O(1)

题目重解给定一个仅包含'L'和'R'的字符串,要求将其分割成尽可能多的子串,且每个子串中'L'和'R'的数量相等。例如输入"R...

发表评论

访客

看不清,换一张

◎欢迎参与讨论,请在这里发表您的看法和观点。