题目:
给你一个按 非递减顺序 排序的整数数组 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 已按 非递减顺序 排序
进阶:
请你设计时间复杂度为 O(n) 的算法解决本问题
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/squares-of-a-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
结果:
解题思路:
首先找到0的位置,也就是负数和正数的交接处,最后一个负数和第一个正数就是 i,j 的取值。
0的数据直接就存储到返回的arr中。
然后就是通过绝对值比较,选择小的取平方放入arr中。
代码:
/**
* Note: The returned array must be malloced, assume caller calls free().
*/
int* sortedSquares(int* nums, int numsSize, int* returnSize){
*returnSize = numsSize;
//先对nums进行排序,先找到0的中介点。
int i = -1, j = 0;
int index = 0;
int * arr = (int *)malloc(sizeof(int) * numsSize);
for(; j < numsSize; j++) {
if(nums[j] > 0) break;
if(nums[j] < 0) i = j;
if(nums[j] == 0) {
arr[index++] = 0;
}
}
if(numsSize == 1) {
arr[0] = nums[0] * nums[0];
return arr;
}
for(; i >= 0 || j < numsSize; ) {
if(i == -1) {
arr[index++] = nums[j] * nums[j];
j++;
}
else if(j == numsSize) {
arr[index++] = nums[i] * nums[i];
i--;
}
else if(abs(nums[i]) < abs(nums[j])) {
arr[index++] = nums[i] * nums[i];
i--;
}
else if(abs(nums[i] == abs(nums[j]))) {
arr[index++] = nums[i] * nums[i];
arr[index++] = arr[index - 2];
i--;
j++;
}
else {
arr[index++] = nums[j] * nums[j];
j++;
}
}
return arr;
}