977有序数组的平方(c++版)
链接: 力扣题目链接
思路
首先要理解题目里的非递减顺序是什么意思。
非递减顺序排列即是指:意思是关键字递增序排列,但是并非单调递增(因为有重复的关键字)从小到大或者允许中间有相等的情形。
比如:5 4 4 3 2 1
1,2,3,4,5 递增排列
5,4,3,2,1 递减排列
1,2,2,3,4,5,5 非递减排列
5,4,3,3,2,1 非递增排列
暴力排序
先将每个数进行平方,再进行排序。不做过多解释。
class Solution
{
public:
vector<int> sortedSquares(vector<int> &nums)
{
for (int i = 0; i < nums.size(); i++)
{
nums[i] *= nums[i];//平方
}
sort(nums.begin(), nums.end()); // 快速排序
return nums;
}
};
注意vector包含在头文件vector中,sort() 函数位于algorithm头文件中。
这个时间复杂度是 O(n + nlogn)
双指针法
数组因为存在负数,所以负数平方后可能大于正数。
该数组平方后是数组的两边值大,两边往中间数值会减少的顺序。
最大值只能出现在数组的两端。
此时可以考虑双指针法了,一个指向数组起始位置,另一个指向终止位置。
定义一个新数组result,和原来数组一样的大小,让指针指向result数组终止位置。
对原数组两边的平方进行比较,将大的数赋值给新数组终止位置。
class Solution
{
public:
vector<int> sortedSquares(vector<int> &nums)
{
int k = nums.size() - 1; //新数组终止位值
vector<int> result(nums.size(), 0); //和原数组一样大小的数组
for (int i = 0, j = nums.size() - 1; i <= j;)
{ // 注意这里要i <= j,因为最后要处理两个元素
if (nums[i] * nums[i] < nums[j] * nums[j]) //最大值在数组右边
{
result[k--] = nums[j] * nums[j]; //注意新数组的终止位置要减1
j--; //原数组终止位置向前移一位
}
else
{
result[k--] = nums[i] * nums[i]; //最大值在数组左边
i++; //原数组起始位置向后移一位
}
}
return result;
}
};
时间复杂度为O(n)
附代码提交结果: