977. 有序数组的平方

https://leetcode.cn/problems/squares-of-a-sorted-array/submissions/562701078/

题目描述

给你一个按 非递减顺序 排序的整数数组 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) 的算法解决本问题

解题思路

直接暴力遍历,数据搞成平方后,进行排序

代码编写

public int[] sortedSquares(int[] nums) {  
	return Arrays.stream(nums).map(num -> num * num).sorted().toArray();  
}

单元测试

System.out.println(Arrays.toString(new Solution977().sortedSquares(new int[]{-4, -1, 0, 3, 10})));  
System.out.println(Arrays.toString(new Solution977().sortedSquares(new int[]{-7, -3, 2, 3, 11})));

额外补充

todo 待优化为o(n)