leetcode 有序数组的平方

cpp 复制代码
class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
        int n = nums.size();
        vector<int> result(n);
        int k = nums.size()-1;
        int left = 0;
        int right = nums.size() - 1;
        
        while ( left <= right ) {
            if ( nums[left]*nums[left] >= nums[right]*nums[right] ) {
                result[k--] = nums[left]*nums[left];
                left++;
            } else {
                result[k--] = nums[right]*nums[right];
                right--;
            }
        }

        return result;
    }
};

思路类似于归并排序,我们需要首先注意到,距离零更远的数字,是平方更大的数字,换句话说,就是绝对值更大的数字,是平方更大的数字。我们这样做是一个线性的做法,如果我们按照绝对值排序,只要是基于比较的排序算法,时间复杂度的下界就是 O(nlogn)O(nlogn)O(nlogn) ,所以的话,我们需要更小的时间复杂度需要考虑不进行排序,考虑用双指针实现我们的需求。

左边的指针 left 从左边往右边扫描,右边的指针 right 从右边往左边扫描。计算对应的元素的平方,取较大者存入我们的结果向量 result ,然后我们是找到的最大的平方,所以的话,我们选择从最大的下标 nums.size()−1nums.size()-1nums.size()−1 开始减小,自减即可。扫描结束,我们的单调非降的向量 result 就完成了。

通过上面的步骤,可以在线性的时间内,把一个有序的数组,平方之后,仍然是有序的数组。双指针的优势非常明显。

相关推荐
李日灐2 天前
【优选算法3】二分查找经典算法面试题
开发语言·c++·后端·算法·面试·二分查找·双指针
小肝一下2 天前
每日两道力扣,day7
数据结构·c++·算法·leetcode·双指针·hot100·接雨水,四数之和
小肝一下3 天前
每日两道力扣,day6
数据结构·c++·算法·leetcode·双指针·hot100
Byte不洛5 天前
LeetCode双指针经典题
c++·算法·leetcode·双指针
Fcy6486 天前
算法基础详解(4)双指针算法
开发语言·算法·双指针
Byte不洛12 天前
LeetCode中经典双指针题(环形链表 + 快乐数 + 移动零)
算法·leetcode·链表·数组·双指针
老四啊laosi18 天前
[双指针] 2. 力扣--复写零
算法·leetcode·双指针·复写零
汉克老师18 天前
GESP2026年3月认证C++五级( 第三部分编程题(2)找数)
c++·排序·双指针·二分算法·gesp5级·gesp五级
伟大的车尔尼20 天前
双指针题目:满足条件的子序列数目
二分查找·排序·双指针
旖-旎22 天前
二分查找(山脉数组的峰顶索引)(5)
c++·算法·leetcode·二分查找·力扣·双指针