LeetCode977 有序数组的平方

172 阅读1分钟

「这是我参与2022首次更文挑战的第17天,活动详情查看:2022首次更文挑战」。

题目描述:

977. 有序数组的平方 - 力扣(LeetCode) (leetcode-cn.com)

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

示例一

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

示例二

输入: nums = [-7,-3,2,3,11]
输出: [4,9,9,49,121]

提示:

  • 1 <= nums.length <= 10^4
  • -104 <= nums[i] <= 10^4
  • nums 已按 非递减顺序 排序

进阶:

  • 请你设计时间复杂度为 O(n) 的算法解决本问题

思路分析

排序

这个就是最简单的了,没有任何花哨的技巧,直接平方之后排序即可。

AC代码

class Solution {
    fun sortedSquares(nums: IntArray): IntArray {
        return nums.map { it * it }.sorted().toIntArray()
    }
}

双指针

我们可以使用两个指针分别指向位置 0 和 n−1,每次比较两个指针对应的数,选择较大的那个逆序放入答案并移动指针。

AC代码

class Solution {
    public int[] sortedSquares(int[] nums) {
        int n = nums.length;
        int[] ans = new int[n];
        for (int i = 0, j = n - 1, pos = n - 1; i <= j;) {
            if (nums[i] * nums[i] > nums[j] * nums[j]) {
                ans[pos] = nums[i] * nums[i];
                ++i;
            } else {
                ans[pos] = nums[j] * nums[j];
                --j;
            }
            --pos;
        }
        return ans;
    }
}

总结

暴力法没什么好说的,双指针法可以好好领会下他的精髓所在。

参考

有序数组的平方 - 有序数组的平方 - 力扣(LeetCode) (leetcode-cn.com)

一行 - 有序数组的平方 - 力扣(LeetCode) (leetcode-cn.com)

kotlin-[有序数组的平方] - 有序数组的平方 - 力扣(LeetCode) (leetcode-cn.com)

各种排序+双指针 - 有序数组的平方 - 力扣(LeetCode) (leetcode-cn.com)