LeetCode977 有序数组的平方

最新推荐文章于 2024-10-02 22:45:05 发布

红毛乌龟

最新推荐文章于 2024-10-02 22:45:05 发布

阅读量1.1k

点赞数 24

分类专栏：算法刷题文章标签：算法 leetcode c++

本文链接：https://blog.csdn.net/daishabby2486/article/details/140667775

版权

算法刷题专栏收录该内容

90 篇文章 0 订阅

订阅专栏

前言

题目： 977.有序数组的平方
文档： 代码随想录——有序数组的平方
编程语言： C++
解题状态： 完成，暴力解题

思路

最简单的思路就是平方以后排序，其中平方的过程的时间复杂度为 $O (n)$ ，整体的时间复杂度基本上取决于排序的时间复杂度。

其次还可以考虑双指针法。

代码

方法一：暴力排序

这里我使用的排序法为时间复杂度 $O(n^2)$ 的选择排序法。

时间复杂度： $O(n^2)$
空间复杂度： $O (1)$

class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
        for (int i = 0; i < nums.size(); i++) {
            nums[i] = nums[i] * nums[i];
        }

        // 选择排序
        for (int i = 0; i < nums.size(); i++) {
            int minIndex = i;
            for (int j = i; j < nums.size(); j++) {
                if (nums[j] < nums[minIndex]) minIndex = j;
            }
            swap(nums[i], nums[minIndex]);
        }

        return nums;
    }
};

方法二：双指针法

数组其实是有序的，只不过负数平方之后可能成为最大数了。

那么数组平方的最大值就在数组的两端，不是最左边就是最右边，不可能是中间。

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

class Solution {
public:
    vector<int> sortedSquares(vector<int>& nums) {
        int k = nums.size() - 1;
        vector<int> result(nums.size(), 0);

        for (int i = 0, j = nums.size() - 1; i <= j;) {
            if (nums[i] * nums[i] > nums[j] * nums[j]) {
                result[k--] = nums[i] * nums[i];
                i++;
            } else {
                result[k--] = nums[j] * nums[j];
                j--;
            }
        }

        return result;
    }
};