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 天前
hot 100 42. 接雨水
数据结构·算法·leetcode·动态规划··双指针
羑悻的小杀马特2 天前
LeetCode 42接雨水全解:暴力超时→DP降维打击→双指针极限压缩空间→单调栈栈式凹槽定位,全景式解析算法优化路径
算法·leetcode·职场和发展·动态规划·双指针·单调栈·接雨水
无尽的罚坐人生7 天前
hot 100 283. 移动零
数据结构·算法·双指针
伟大的车尔尼7 天前
双指针题目:K 和数对的最大数目
双指针·哈希表
长安er12 天前
LeetCode136/169/75/31/287 算法技巧题核心笔记
数据结构·算法·leetcode·链表·双指针
伟大的车尔尼16 天前
双指针题目:两数之和 IV - 输入二叉搜索树
二叉树··二叉搜索树·双指针
yaoh.wang17 天前
力扣(LeetCode) 88: 合并两个有序数组 - 解法思路
python·程序人生·算法·leetcode·面试·职场和发展·双指针
yaoh.wang20 天前
力扣(LeetCode) 27: 移除元素 - 解法思路
python·程序人生·算法·leetcode·面试·职场和发展·双指针
伟大的车尔尼20 天前
双指针题目:两个数组的交集 II
排序·双指针·哈希表