【leetcode】977. 有序数组的平方（squares-of-a-sorted-array）（双指针）[简单]

KroneX

于 2020-10-16 23:51:26 发布

阅读量257

点赞数

分类专栏： leetcode 题解双指针文章标签： leetcode 双指针

本文链接：https://blog.youkuaiyun.com/Krone_/article/details/109126590

版权

题解同时被 3 个专栏收录

217 篇文章

订阅专栏

leetcode

215 篇文章

订阅专栏

双指针

9 篇文章

订阅专栏

该博客介绍了如何解决LeetCode上的一个问题——给定一个非递减排序的整数数组，返回每个数字平方后的数组，同样保持非递减排序。博主分享了他们在解决此问题时的思路，包括考虑正负数的情况，使用双指针的方法，以及不同情况下的处理方式。最终给出了AC代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

链接

https://leetcode-cn.com/problems/squares-of-a-sorted-array/

耗时

解题：9 min
题解：18 min

题意

给定一个按非递减顺序排序的整数数组 A，返回每个数字的平方组成的新数组，要求也按非递减顺序排序。

思路

刚开始没有想清楚，直接上手写的，所以我的思路有点复杂 😦 。非常不建议用。

数组中有正有负的时候，找到正负分割的地方，或者 0 的位置。然后用双指针。
数组中元素非正，平方倒放。
数组中元素非负，平方正放。

时间复杂度： $O (n)$

AC代码

class Solution {
public:
    vector<int> sortedSquares(vector<int>& A) {
        vector<int> res;
        int n = A.size();
        if(A[0] < 0 && A[n-1] > 0) {
            int p, q;
            for(int i = 1; i < n; ++i) {
                if(A[i] == 0) {
                    p = i-1;
                    q = i+1;
                    res.push_back(0);
                    break;
                }
                if(A[i-1]*A[i] < 0) {
                    p = i-1;
                    q = i;
                    break;
                }
            }
            while(p >= 0 && q < n) {
                if(A[p]*A[p] <= A[q]*A[q]) {
                    res.push_back(A[p]*A[p]);
                    p--;
                }
                else {
                    res.push_back(A[q]*A[q]);
                    q++;
                }
            }
            while(p >= 0) {
                res.push_back(A[p]*A[p]);
                p--;
            }
            while(q < n) {
                res.push_back(A[q]*A[q]);
                q++;
            }
        }
        else if(A[0] < 0 && A[n-1] <= 0) {
            for(int i = n-1; i >= 0; --i) {
                int a = A[i];
                res.push_back(a*a);
            }
        }
        else {
            for(auto a : A) {
                res.push_back(a*a);
            }
        }
        return res;
    }
};