给你一个按 非递减顺序 排序的整数数组 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 已按 非递减顺序 排序
来源:力扣(LeetCode)
方法——双指针
思路与算法:
显然的,如果数组中所有的数都是非负数,那么平方过后不用排序,数组仍是升序;如果数组中的数全是负数,那么平方过后的数组是降序。
这里我们可以找到数组中负数与非负数的分界线,将其设为neg,也就是说,nums[0] - nums[neg]是负数,nums[neg + 1] - nums[numsSize - 1]是非负数。当我们将数组平方之后,从nums[0]到nums[neg]都是单调递减的,从nums[neg + 1]到nums[numsSize - 1]是单调递增的。
这样我们就得到了两个有序的子数组,接下来,用两个指针分别指向neg和neg+1,每次比较两个指针所指的数,小的那一个放进答案数组中,小的那一个对应的指针向边界移动,!](https://img-blog.csdnimg.cn/bb2618a85971432c850b9665ec3de1ca.jpeg#pic_center)
直到某个指针到了边界,将另一个指针还没遍历到的数(剩下的那些数)直接放进答案数组中即可。
下面是代码
int* sortedSquares(int* nums, int numsSize, int* returnSize) {
int negative = -1;
for (int i = 0; i < numsSize; ++i) {
if (nums[i] < 0) {
negative = i;
} else {
break;
}
}
int* ans = malloc(sizeof(int) * numsSize);
*returnSize = 0;
int i = negative, j = negative + 1;
while (i >= 0 || j < numsSize) {
if (i < 0) {
ans[(*returnSize)++] = nums[j] * nums[j];
++j;
} else if (j == numsSize) {
ans[(*returnSize)++] = nums[i] * nums[i];
--i;
} else if (nums[i] * nums[i] < nums[j] * nums[j]) {
ans[(*returnSize)++] = nums[i] * nums[i];
--i;
} else {
ans[(*returnSize)++] = nums[j] * nums[j];
++j;
}
}
return ans;
}