LeetCode 977-有序数组的平方

最新推荐文章于 2024-11-15 22:13:14 发布

raysy

最新推荐文章于 2024-11-15 22:13:14 发布

阅读量154

点赞数

分类专栏： LeetCode 文章标签： leetcode 算法

本文链接：https://blog.csdn.net/qq_45820252/article/details/120174435

版权

LeetCode 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

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

示例 1：
输入：nums = [-4,-1,0,3,10]
输出：[0,1,9,16,100]
解释：平方后，数组变为 [16,1,0,9,100]
排序后，数组变为 [0,1,9,16,100]
示例 2：
输入：nums = [-7,-3,2,3,11]
输出：[4,9,9,49,121]

提示：
1 <= nums.length <= 104
-104 <= nums[i] <= 104
nums 已按非递减顺序排序

进阶：请你设计时间复杂度为 O(n) 的算法解决本问题
来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/squares-of-a-sorted-array

方法一:
查找临界点,因为已经是按照递增的顺序排序,按照临界点,右边按照顺序的方式排列,左边按照逆序的方式排列,这样就可以用归并的思想来完成,找出最后一个负数为临界点,如果临界点的平方大于临界点+1的平方,那么临界点+1最为最小值,将它填入到临时数组的起点,然后临时数组的下标+1,因为临界点最左边的数的平方都比临界点大,所以向右移动临界点+1的下标继续比较;而如果临界点的平方小于临界点+1的平方,就将临界点添加到数组中,然后将临界点向左移动继续比较;
如果临界点不存在,说明数组全是正数,直接平方填入即可;
如果临界点=数组长度-1,说明全是负数,按照逆序填入平方即可

    public static int[] sortedSquares(int[] nums) {
        int n = nums.length;
        int[] result = new int[nums.length];
        int side=-1;
        for(int i=0;i<n;++i){
            if(nums[i]<0){
                side=i;
            }else{
                break;
            }
        }
        int index=0,i=side,j=side+1;
        while(i>=0||j<n){
            if(i<0){
                result[index]=nums[j]*nums[j];
                ++j;
            }else if(j==n){
                result[index]=nums[i]*nums[i];
                --i;
            }else if(nums[i]*nums[i]<=nums[j]*nums[j]){
                result[index]=nums[i]*nums[i];
                --i;
            }else{
                result[index]=nums[j]*nums[j];
                ++j;
            }
            ++index;
        }
        return result;
    }

方法二:
用双指针的方法来收缩下标,如果有正有负,那么数组最大值一定存在两端,比较最大值,如果left>rigth,就将nums[left]平方填入到临时数组的最高位,然后left++向右移动,如果小于,就将nums[right]的平方值填入到临时数组的最高位,然后right–向左移动,同时两者都要将临时数组的指针向左移动,一直查找直到left>right为止.
如果全为正或者负,那么就其中一个指针移动直到两个指针重合,同样从临时数组的最高位开始填.

    public static int[] sortedSquares2(int[] nums) {
        int left = 0;
        int right = nums.length-1;
        int[] result=new int[right+1];
        int write = right;
        while(left<=right){
            if(nums[left]*nums[left]>nums[right]*nums[right]){
                result[write]=nums[left]*nums[left];
                ++left;
            }else{
                result[write]=nums[right]*nums[right];
                --right;
            }
            --write;
        }
        return result;
    }