算法——第二天(双指针)

给你一个按 非递减顺序 排序的整数数组 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 已按 非递减顺序 排序

来源:力扣(LeetCode)

方法——双指针
思路与算法:
显然的,如果数组中所有的数都是非负数,那么平方过后不用排序,数组仍是升序;如果数组中的数全是负数,那么平方过后的数组是降序。

这里我们可以找到数组中负数与非负数的分界线,将其设为neg,也就是说,nums[0] - nums[neg]是负数,nums[neg + 1] - nums[numsSize - 1]是非负数。当我们将数组平方之后,从nums[0]到nums[neg]都是单调递减的,从nums[neg + 1]到nums[numsSize - 1]是单调递增的。

这样我们就得到了两个有序的子数组,接下来,用两个指针分别指向neg和neg+1,每次比较两个指针所指的数,小的那一个放进答案数组中,小的那一个对应的指针向边界移动,!](https://img-blog.csdnimg.cn/bb2618a85971432c850b9665ec3de1ca.jpeg#pic_center)
直到某个指针到了边界,将另一个指针还没遍历到的数(剩下的那些数)直接放进答案数组中即可。
这是我解题的时候写的

下面是代码

int* sortedSquares(int* nums, int numsSize, int* returnSize) {
    int negative = -1;
    for (int i = 0; i < numsSize; ++i) {
        if (nums[i] < 0) {
            negative = i;
        } else {
            break;
        }
    }

    int* ans = malloc(sizeof(int) * numsSize);
    *returnSize = 0;
    int i = negative, j = negative + 1;
    while (i >= 0 || j < numsSize) {
        if (i < 0) {
            ans[(*returnSize)++] = nums[j] * nums[j];
            ++j;
        } else if (j == numsSize) {
            ans[(*returnSize)++] = nums[i] * nums[i];
            --i;
        } else if (nums[i] * nums[i] < nums[j] * nums[j]) {
            ans[(*returnSize)++] = nums[i] * nums[i];
            --i;
        } else {
            ans[(*returnSize)++] = nums[j] * nums[j];
            ++j;
        }
    }

    return ans;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值