一、有序数组的平方
题目链接:https://leetcode.cn/problems/squares-of-a-sorted-array/comments/
看到题目首先的想法是暴力解,其次是利用双指针。先说暴力解:直接计算平方然后给数组进行排序。这里就不忘容器里面塞了,想要练习一下快排。
快排的时间复杂度是O(nlog2n),但是在有序的时候时间复杂度最高达到O(n2)。
其基本思想就是选取一个基准元素(非递减排序):一趟排序下来让所有比它大的元素放在它的右边,所有比它小的元素放在它的左边,即一趟排序就能找到这个基准元素的最终位置。具体实现
1.从右往左找,找到最右边元素小于基准元素的元素所在的位置
2.从左往右找,找到最左边元素小于基准元素的元素所在的位置
3.交换这两个元素
4.重复上述过程直到左边的指针与右边的指针相遇。
上述过程完成了一趟排序,想要完成多次排序,就需要选取多次基准,这体现的是“分治”的思想,需要用递归方式来实现。
class Solution {
public int[] sortedSquares(int[] nums) {
for(int i = 0;i < nums.length;i++){
nums[i] = nums[i] * nums[i];
}
sortNums(nums,0,nums.length-1);
return nums;
}
void sortNums(int[] nums,int start,int end){
if(start >= end){
return;
}
int left = start,right = end,temp = 0;
int val = nums[left];
while(left < right){
while(left < right && nums[right] >= val){
right--;
}
while(left < right && nums[left] <= val){
left++;
}
temp = nums[left];
nums[left] = nums[right];
nums[right] = temp;
}
temp = nums[left];
nums[left] = val;//写right也行,因为此时左右指针相遇了,即找到了val元素真正的位置
nums[start] = temp;
//以关键字为中心划分左右区间,然后在这两个区间里面递归调用
sortNums(nums,start,left-1);
sortNums(nums,right+1,end);
}
}
有一说一,快排想起来算法很简单,但是做起来好难,连学带写带找bug搞了快一个半小时,真的离谱。有点后悔我闲的没事练什么快排,难搞。
接下来还是正解这道题吧,用双指针:
思路就是给的数组是有序的,要把这个特点利用起来,平方以后最大值总是出现在两边。所以就是定义两个指针,一个指针left,一个指针right。比较这两个指针所指的值,来确定最大值。
找到最大值以后,将最大值放入新的数组,如果这个最大值是从左边取到的就left++,如果是右边取到的就right–。上代码:
class Solution {
public int[] sortedSquares(int[] nums) {
for(int i = 0;i < nums.length;i++){
nums[i] = nums[i] * nums[i];
}
int left = 0,right = nums.length-1;
int k = nums.length-1;
int[] finalNums = new int[nums.length];
for(int i = left,j = right;i <= j;){
if(nums[i] > nums[j]){
//移动左边的指针
finalNums[k] = nums[i];
i++;
k--;
}
else{
//当最大值在右边的时候,移动右边的指针,当相等的时候两个都是最大值,无所谓选左边还是选右边
finalNums[k] = nums[j];
j--;
k--;
}
}
return finalNums;
}
}