代码随想录之有序数组平方排序

本文介绍了如何解决LeetCode的一道算法题,要求根据一个非递减数组生成其平方后的非递减数列。通过设置双指针,从数组的首尾开始,比较平方值并填充结果数组,确保了排序不变。代码分别用Java和Python实现,思路清晰,效率较高。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

 此文为阅读’代码随想录‘的leetcode刷题所写笔记。代码参考作者’程序员Carl‘(代码随想录)

参考链接:代码随想录

 力扣题目链接

已知一个非递减数组, 要求返回该数组平方后的非递减数列.

分析

该数组的长度已知, 我们可以设置两个指针分别指向首和尾, 并向中间移动.

由于数组是有序的, 因此平方后最大的只会在两侧的其中一个, 因此可以每次在两端比较选择.

代码实现

Java

class Solution{
    public int[] Result(int[] nums){
        int n = num.length;
        int i = 0;
        int j = n-1;
        int[] result;
        
        while(i<=j){
            if (nums[i]*nums[i] <= nums[j]*nums[j]){
                result[n--] = nums[j]*nums[j];
                j--;
            }else {
                result[n--] = nums[i]*nums[i];
                i++;
            }
        }
        return result;
    }
}
    

Python

def solution(nums: List[int])-> List[int]:
    n = len(nums)
    i,j,k = 0,n-1,0
    result = [-1]*n
    while i<=j:
        if nums[i]**2 <= nums[j]**2:
            result[k] = num[i]**2
            i+=1
            k+=1
        else:
            result[k] = num[j]**2
            j+=1
            k+=1
    return result

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值