Leetcode977. 有序数组的平方

Every day a leetcode

题目来源：977. 有序数组的平方

解法1：排序

遍历一次数组，每个元素取平方后排序。

代码：

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */

int cmpfunc (const void * a, const void * b)
{
   return ( *(int*)a - *(int*)b );
}

int* sortedSquares(int* nums, int numsSize, int* returnSize){
    *returnSize=numsSize;
    int *ans;
    ans=(int*)malloc(numsSize*sizeof(int));
    for(int i=0;i<numsSize;i++)
    {
        ans[i]=nums[i]*nums[i];
    }
    qsort(ans,numsSize,sizeof(int),cmpfunc);
    return ans;
}

结果：

解法2：双指针

我们发现：
在数组 nums 中，每个元素取平方后，其中的最大值一定在最左侧或最右侧取得。

于是，我们可以设置两个指针，分别指向 0 和 numSize-1 。每次比较两个指针对应的数，选择较大的那个逆序放入答案数组 ans 并移动指针 pos 。

代码：

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* sortedSquares(int* nums, int numsSize, int* returnSize){
    *returnSize=numsSize;
    int *ans;
    ans=(int*)malloc(numsSize*sizeof(int));
    int i=0;
    int j=numsSize-1;
    int pos=numsSize-1;
    int a,b;
    while(i<=j)
    {
        a=nums[i]*nums[i];
        b=nums[j]*nums[j];
        if(a>b)
        {
            ans[pos]=a;
            i++;
        }
        else
        {
            ans[pos]=b;
            j--;
        }
        pos--;
    }
    return ans;
}

结果：