力扣977 有序数组(双指针)
给你一个按 非递减顺序 排序的整数数组 nums,返回 每个数字的平方 组成的新数组,要求也按 非递减顺序 排序。
示例 1:
输入:nums = [-4,-1,0,3,10]
输出:[0,1,9,16,100]
解释:平方后,数组变为 [16,1,0,9,100]
排序后,数组变为 [0,1,9,16,100]
思路:最大的在两边,最小的在中间。因此i从左往右,j从右往左。注意当i==j时也要继续遍历把中间的值加上,所以要是i<=j
class Solution {
public:
vector<int> sortedSquares(vector<int>& nums) {
int len=nums.size();
vector<int> vec(nums);
int i=0,j=len-1;
int t=len-1;
for(i,j;i<=j;){
if(nums[i]*nums[i]<nums[j]*nums[j]){
vec[t--]=nums[j]*nums[j];
j--;
}
else{
vec[t--]=nums[i]*nums[i];
i++;
}
}
return vec;
}
};
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/squares-of-a-sorted-array