LeetCode——977.有序数组的平方【双指针】

leetcode——977.有序数组的平方（双指针）

---

问题描述

给你一个按非递减顺序排序的整数数组 nums，返回每个数字的平方组成的新数组，要求也按非递减顺序排序。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/squares-of-a-sorted-array

1.1 示例

输入：nums = [-4,-1,0,3,10]
输出：[0,1,9,16,100]
解释：平方后，数组变为 [16,1,0,9,100]
排序后，数组变为 [0,1,9,16,100]

输入：nums = [-7,-3,2,3,11]
输出：[4,9,9,49,121]

1.2 提示

• 1 <= nums.length <= 104
• -104 <= nums[i] <= 104
• nums 已按非递减顺序排序

代码

方法一 直接排序

int cmp(const void* _a, const void* _b) {
    int a = *(int*)_a, b = *(int*)_b;
    return a - b;
}

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

方法二 双指针（两种）

• 思路：使用两个指针分别指向位置 0和 n-1，每次比较两个指针对应的数，选择较大的那个逆序放入数组中并移动指针。

int* sortedSquares(int* nums, int numsSize, int* returnSize) {
    int* ans = malloc(sizeof(int) * numsSize);
    *returnSize = numsSize;  //返回的数组大小就是原数组大小
    for (int i = 0, j = numsSize - 1, pos = numsSize - 1; i <= j;) {
        if (nums[i] * nums[i] > nums[j] * nums[j]) {
            ans[pos] = nums[i] * nums[i];
            ++i;
        } 
        else {
            ans[pos] = nums[j] * nums[j];
            --j;
        }
        --pos;
    }
    return ans;
}

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* sortedSquares(int* nums, int numsSize, int* returnSize){
    *returnSize = numsSize;   //返回的数组大小就是原数组大小

    //创建两个指针，left指向数组第一位元素，right指向数组最后一位元素
    int left = 0,right = numsSize - 1;

    //最后要返回的结果数组
    int* ans = (int*)malloc(sizeof(int) * numsSize);
    int index;
    for(index = numsSize - 1 ; index >= 0 ; index--){
        int lSquare = nums[left] * nums[left];  //左指针指向元素的平方
        int rSquare = nums[right] * nums[right];  //右指针指向元素的平方
        if(lSquare > rSquare){
        //若左指针指向元素平方比右指针指向元素平方大，将左指针指向元素平方放入结果数组。左指针右移一位
            ans[index] = lSquare;
            left++;
        }
        else {
        //若右指针指向元素平方比左指针指向元素平方大，将右指针指向元素平方放入结果数组。右指针左移一位
            ans[index] = rSquare;
            right--;
        }
    }
    return ans;  //返回结果数组
}