题目:
给你一个按 非递减顺序 排序的整数数组 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]
解题思路:
使用双指针法:数组其实是有序的, 只不过负数平方之后可能成为最大数了。那么数组平方的最大值就在数组的两端,不是最左边就是最右边,不可能是中间。
此时可以考虑双指针法了,i指向起始位置,j指向终止位置。
定义一个新数组result,和nums数组一样的大小,让k指向result数组终止位置。
如果nums[i] * nums[i] < nums[j] * nums[j] 那么result[k- -] = nums[j] * nums[j];
如果nums[i] * nums[i] >= nums[j] * nums[j] 那么result[k- -] = nums[i] * nums[i];
题解
代码如下(示例):
#include<iostream>
#include<vector> //包含vector的头文件
using namespace std;
vector<int>square(vector<int>nums)
{
int i = 0;
int j = nums.size() - 1;
vector<int>result(nums.size(),0);
for(int k = nums.size()-1;i<=j;)
{
if(nums[i]*nums[i] < nums[j]*nums[j])
{
result[k--] = nums[j]*nums[j];
j--;
}
else if(nums[i]*nums[i] >= nums[j]*nums[j])
{
result[k--] = nums[i]*nums[i];
i++;
}
}
/*返回排序和平方后的数组*/
return result;
}
int main()
{
vector<int> V = {-7,-4,-1,0,3,5,8}; //假定输入的数组为V
V = square(V);
cout << "Elements in the vector after square: " ;
for(int i : V) cout << i << " ";
cout << endl;
}
运行结果
总结
双指针法的时间复杂度为O(n),相对于暴力排序(也就是先把数组平方然后用sort快速排序的方法)的解法O(n + nlog n)还是提升不少的。