有序数组的平方
给你一个按非递减顺序排序的整数数组 nums,返回每个数字的平方组成的新数组,要求也按非递减顺序排序。
极限情况分析,涉及到div元素
- 只有[0]
- 全为正数,例[1,2,3,4]
- 全为负数[-4,-3,-2,-1]
其他注意事项:
- 要注意将最后剩余的元素写回
/**
* Note: The returned array must be malloced, assume caller calls free().
*/
int Cal_Square(int a,int b){
int a_squ = a*a;
int b_squ = b*b;
if(a_squ>b_squ)
return 1;
else if(a_squ<b_squ)
return -1;
else
return 0;
}
int* sortedSquares(int* nums, int numsSize, int* returnSize){
//1.寻找分界点
int div = 0;
for(int i = 0;i<numsSize;i++){
if(nums[i]<0){
div = i;
}
else break;
}
*returnSize = 0;
int p1 = div,p2=div+1;
//int* ans = malloc(sizeof(int) * numsSize);
int *s = malloc(numsSize*sizeof(int));
while(p1 >= 0 && p2 < numsSize){
if(Cal_Square(nums[p1],nums[p2])==1){
s[(*returnSize)++] = nums[p2]*nums[p2];
p2++;
}
else {
s[(*returnSize)++] = nums[p1]*nums[p1];
p1--;
}
}
while(p1>=0){
s[(*returnSize)++] = nums[p1]*nums[p1];
p1--;
}
while(p2 < numsSize){
s[(*returnSize)++] = nums[p2]*nums[p2];
p2++;
}
return s;
}