【79.有序数组的平方】

67 阅读1分钟

题目

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

示例 1:

输入: nums = [-4,-1,0,3,10]
输出: [0,1,9,16,100]
解释: 平方后,数组变为 [16,1,0,9,100]
排序后,数组变为 [0,1,9,16,100]

题解

方式一:暴力

public int[] sortedSquares(int[] nums) {
    int[] result = new int[nums.length];
    for (int i = 0; i < nums.length; i++) {
        result[i] = nums[i] * nums[i];
    }
    Arrays.sort(result);
    return result;
}

方式二:双指针

原数组有序,所以平方后相对有序

public int[] sortedSquares(int[] nums) {
    int n = nums.length;
    int mix = 0; // 平方最小的下标
    int[] arr = new int[n]; // 原数组平方
    for (int i = 0; i < nums.length; i++) {
        int tem = nums[i] * nums[i];
        mix = tem < arr[mix] ? i : mix;
        arr[i] = tem;
    }

    int[] result = new int[n];
    result[0] = arr[mix];
    int l = mix - 1, r = mix + 1, index = 1;
    while (l >= 0 && r < n) {
        if (arr[l] > arr[r]) {
            result[index++] = arr[r++];
        } else {
            result[index++] = arr[l--];
        }
    }
    while (index < n) {
        if (l < 0) {
            result[index++] = arr[r++];
        } else {
            result[index++] = arr[l--];
        }
    }

    return result;
}

总结

算法:双指针