给你一个按 非递减顺序 排序的整数数组 nums
,返回 每个数字的平方 组成的新数组,要求也按 非递减顺序 排序。
输入:nums = [-4,-1,0,3,10] 输出:[0,1,9,16,100] 解释:平方后,数组变为 [16,1,0,9,100] 排序后,数组变为 [0,1,9,16,100]
最简单的方法就是把这个数组全部平方一遍再来排序,我们可以用双指针的方法,因为原数组是一个从小到大的数组,那么平方后最大的数只会在最前面或者最后面得出,所以我们可以让一个指针指向最前面,一个指针指向数组最后面,让这两个指针指向的元素对比,就可以很容易找出最大的数了,两头的指针向中间靠近,通过多次对比就可以实现平方后的排序,直接看代码
#include<iostream>
using namespace std;
#include<vector>
vector<int> sortedSquares(vector<int>& nums) {//nums = [-4,-1,0,3,10]
int i = 0; //这个指针用于指向数组最前面的元素
int k = nums.size() - 1;//指向数组最末尾的元素的指针,用于构造平方后排序的数组
int j = k; //指向数组最末尾元素的指针
vector<int> result(nums.size(), 0);//平方后排序的数组,先用0填满
while (i <= j) { //两头的指针是向中间靠拢的,因为我们要把原来的数组里的元素的
//平方写到一个新数组里,所以是i<=j,如果是i<j,那么i = j的那个元素就被漏掉了
if (nums[i] * nums[i] > nums[j] * nums[j]) {
result[k] = nums[i] * nums[i];
k--;
i++;
}
else {
result[k] = nums[j] * nums[j];//-4的平方和10的平方对比自然10比较大所以先执行这一 句
k--; //新数组写入元素后,k就要左移了,j也要左移,同理,如果写入的是i所指向的元素,k左移,i就要右移
j--;
}
}
return result;
}
int main() {
vector<int> nums({ -4,-1,0,3,10 });
vector<int> ret = sortedSquares(nums);
for (int i = 0; i < ret.size() ; i++) {
cout << ret[i] << " ";
}
system("pause");
return 0;
}