LeetCode 977. 有序数组的平方（排序、双指针）

岁忧

于 2024-09-08 22:56:16 发布

阅读量46

点赞数 1

分类专栏： LeetCode 文章标签： leetcode 算法职场和发展 c++ 数据结构

本文链接：https://blog.csdn.net/weixin_46028214/article/details/142034605

版权

LeetCode 专栏收录该内容

440 篇文章 0 订阅

订阅专栏

977. 有序数组的平方

在这里插入图片描述

方法一：排序，将平方和的值升序排序。时间复杂度0(nlogn)

class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
        vector<int> v;
        for(auto x:nums){
            v.push_back(x*x);
        }
        sort(v.begin(),v.end());
        return v;
    }
};

方法二：双指针，[0,i]都是负数，[j,n-1]都是正数。时间复杂度0(n)。

class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
        vector<int> v;
        int n=nums.size();
        int i=0;
        for(;i<n;i++){
            if(nums[i]>=0){
                i--;
                break;
            }
        }
        if(i==n) i--;
        int j=i+1;
        while(i>=0||j<n){
            if(i<0){
                v.push_back(nums[j]*nums[j]);
                j++;
            }else if(j==n){
                v.push_back(nums[i]*nums[i]); 
                i--;  
            }else if(nums[i]*nums[i]<nums[j]*nums[j]){
                v.push_back(nums[i]*nums[i]);
                i--;
            }else{
                v.push_back(nums[j]*nums[j]);
                j++;
            }
        }
        return v;
    }
};

方法三：双指针。这次是从找最大的开始。时间复杂度0(n)

class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
        int n=nums.size();
        vector<int> v(n);
        int i=0,j=n-1;
        int t=n-1;
        while(i<=j){
            if(nums[i]*nums[i]<=nums[j]*nums[j]){
                v[t]=nums[j]*nums[j];
                j--;
            }else{
                v[t]=nums[i]*nums[i];
                i++;
            }
            t--;
        }
        return v;
    }
};