力扣第 977.有序数组的平方 笔记
题目建议: 本题关键在于理解双指针思想
题目链接:https://leetcode.cn/problems/squares-of-a-sorted-array/
给你一个按 非递减顺序 排序的整数数组 nums,返回 每个数字的平方 组成的新数组,要求也按 非递减顺序 排序。
示例 1:
输入:nums = [-4,-1,0,3,10]
输出:[0,1,9,16,100]
解释:平方后,数组变为 [16,1,0,9,100]
暴力解法
排序后,数组变为 [0,1,9,16,100]
从题目中看,这道题我首先想到的思路就是,先对数组中的元素进行平方,在进行排序。
class Solution {
public int[] sortedSquares(int[] nums) {
int temp = 0;
boolean flag=false;
for (int i = 0; i <nums.length ; i++) {
nums[i] = nums[i]*nums[i];
}
//冒泡排序
for (int i = 0; i <nums.length-1 ; i++) {
for (int j = 0; j <nums.length-1-i ; j++) {
if (nums[j]>=nums[j+1]){
flag=true;
temp=nums[j];
nums[j]=nums[j+1];
nums[j+1]=temp;
}if (!flag){
break;
}
else{
flag=false;
}
}
}
return nums;
}
}
这个方法由于三个for循环,因此执行时间较长。可以在此基础上进行改进在排序的时候改用Arrays.sort(nums),运行之后,执行耗时5ms,速度不是很快。
双指针
数组是有序数组,对于负数平方之后可能会成为最大值,那么数组平方的最大值就在数组的两端,此时i指向起始位置,j指向终止位置。如果左边小于右边的平方值,那么结果集向左移动一位,j指针左移,i指针不动,如果左边平方值大于右边平方值,那么结果集向左移动,i指针向右移动,j指针向左移动。
//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
public int[] sortedSquares(int[] nums) {
int i = 0;
int j =nums.length-1;
int [] result = new int[nums.length];
int index =result.length-1;
while (i<j){
//左边小于右边
if (nums[i]*nums[i]<nums[j]*nums[j]){
result[index--]=nums[j]*nums[j];
//左移
j--;
}
//右边小于左边
else {
result[index--]=nums[i]*nums[i];
//右移
i++;
}
}
return result;
}
}
//leetcode submit region end(Prohibit modification and deletion)