LeetCode 977-有序数组的平方

给你一个按 非递减顺序 排序的整数数组 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;
    }
上一篇:力扣977. 有序数组的平方


下一篇:力扣977.有序数组的平方