当前位置：首页 > article >正文

【Leetcode刷题随笔】977 有序数组的平方

article 2025/1/5 6:36:25

1. 题目描述

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

示例：

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

2. 解题方法

2.1 方法一：直接排序

最直观的方法，直接平方之后用qsort函数排序。代码实现如下：

//定义比较函数，用于下面qsort函数排序
/*
该函数接收两个指向 const void 的指针（可以指向任何类型的数据），
将它们转换为指向 int 的指针，然后解引用这些指针以获取实际的整数值 a 和 b。
最后，它返回 a - b 的结果。这个返回值决定了排序的顺序：
1.如果返回值小于 0，则 _a 会被放置在 _b 之前。
2.如果返回值等于 0，则 _a 和 _b 的相对位置不变（但具体实现可能有所不同）。
3.如果返回值大于 0，则 _a 会被放置在 _b 之后。
*/
int cmp(const void* _a, const void*_b){
    int a = *(int*) _a;
    int b = *(int*) _b;
    return a - b;
}

int* sortedSquares(int* nums, int numsSize, int* returnSize) {
    *returnSize = numsSize;
    int* res = malloc(sizeof(int) * numsSize);
    for(int i = 0; i < numsSize; i++){
        res[i] = nums[i] * nums[i];
    }
    qsort(res, numsSize, sizeof(int), cmp);//调用qsort函数和上面的比较函数对res进行排序
    return res;
}

2.2 方法二：双指针

对于包含负数的数组来说，最小的负数平方之后可能变为最大。整个数组的最大值只可能在数组最两端出现，不可能在其他位置。此时可以考虑双指针方法。

定义两个指针 i 和 j 分别指向原始数组两端，用于比较大小。再定义一个指针 k 指向结果数组的最右端，用于存放较大值。

如果nums[ i ] * nums[ i ] < nums[ j ] * nums[ j ]，则将 j 指向位置的平方填入结果数组中 k 指向的位置，然后 k 左移，等待下一次填充。j 也左移，继续比较下一个数。
如果nums[ i ] * nums[ i ] >= nums[ j ] * nums[ j ]，则将 i 指向位置的平方填入结果数组中 k 指向的位置，然后 k 左移，等待下一次填充。i 右移，继续比较下一个数。

代码实现如下：

int* sortedSquares(int* nums, int numsSize, int* returnSize) {
    *returnSize = numsSize;
    int* res = malloc(sizeof(int) * numsSize);
    int i = 0, j = numsSize - 1, k = numsSize - 1;
    while(i <= j){
        int i_sqr = nums[i] * nums[i];
        int j_sqr = nums[j] * nums[j];
        if(i_sqr < j_sqr){
            res[k] = j_sqr;
            k--;
            j--;
        }else{
            res[k] = i_sqr;
            k--;
            i++;
        }
    }
    return res;
}