[Day 10]有序数组的平方
今天这道题较之前的思路会好理解些,题目链接:977.有序数组的平方
题目描述:
给你一个按 非递减顺序 排序的整数数组 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 已按 非递减顺序 排序
暴力解法
这道题最直接了当的理解,给定了一个数组,把它平方后再排序。
代码如下:
class Solution {
public:
vector<int> sortedSquares(vector<int>& nums) {
for(int i=0;i<nums.size();i++){
nums[i]=nums[i]*nums[i];//将原来的数组平方后更新为新的数组nums
}
sort(nums.begin(),nums.end());//将nums从头到尾 进行快速排序
return nums;//返回平方排序后的数组
}
};
双指针法
思路
本道题可以用双指针i,j定义为数组的开始和末尾元素,负数平方后要么是最小要么最大,不会为0。定义一个和nums同样大小的数组用来记录平方排序后的新数组,因为是非递减平方排序后的数组,最大值在末尾记录为k,逆序把原来的数组处理后往新数组赋值。
如果nums[i]*nums[i]<nums[j]*nums[j],把nums[j]*nums[j]赋值给result的k,k–。
如果nums[i]*nums[i]>=nums[j]*nums[j], 把 nums[i]*nums[i]=值给result的k,k–。
理清思路,代码也就很好写出来了。
1.定义一个新数组和nums一样大
vector<int> result(nums.size(), 0);//定义result,大小和nums一样大,元素都初始化为0
2.定义k使得原数组比较后将大的数插入新数组的最大值
int k = result.size() - 1;//定义k为末尾元素
3.使用一个for循环遍历两个指针,当头指针i和尾指针j移动到i>j的时候证明nums数组已经全排序好赋值给result了。
for (int i = 0, j = nums.size() - 1; i <= j;) //注意这里i<=j是吧两个指针都遍历完成后才结束循环
4. 如果nums[i]*nums[i]<nums[j]*nums[j],把nums[j]*nums[j]赋值给result的k,k–。j指针向前移动一位。
if (nums[i] * nums[i] < nums[j] * nums[j]) {
result[k] = nums[j] * nums[j];
k--;
j--;
}
5.如果nums[i]*nums[i]>=nums[j]*nums[j],把nums[i]*nums[i]赋值给result的k,k–。i指针向后移动一位。
result[k] = nums[i] * nums[i];
k--;
i++;
6.循环结束,返回新的数组
return result;
完整代码:
class Solution {
public:
vector<int> sortedSquares(vector<int>& nums) {
vector<int> result(nums.size(), 0);//定义一个和旧数组一样大的新数组,并初始化为0
int k = result.size() - 1;//定义k为新数组末尾元素,用于比较新旧数组的大值
for (int i = 0, j = nums.size() - 1; i <= j;) {//使用双指针来遍历旧数组,一直到i>j时,旧数组遍历完成
if (nums[i] * nums[i] < nums[j] * nums[j]) {//当i^2小于j^2
result[k] = nums[j] * nums[j];//赋值给k
k--;//k向前移,进行下一个赋值
j--;//j也向前移动
} else {
result[k] = nums[i] * nums[i];//如果条件相反,执行下面
k--;
i++;
}
}
return result;//循环结束,返回新数组
}
};