【刷题笔记day2】左右指针、滑动窗口、模拟

最新推荐文章于 2024-04-29 09:11:57 发布

VIP文章轰车车

最新推荐文章于 2024-04-29 09:11:57 发布

阅读量543

点赞数 2

分类专栏： LeetCode刷题笔记文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/polycozy/article/details/126990506

版权

左右指针

LC977 有序数组的平方

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

class Solution:
    def sortedSquares(self, nums: List[int]) -> List[int]:
        n = len(nums)
        # 初始化左右指针，一头一尾
        i, j = 0, n - 1
        # 结果数组的写入位置的指针，逆序写入
        p = n - 1
        res = [0] * n
        # 执行双指针合并有序数组的逻辑
        # 由于逆序写入，绝对值大的先写入，并且指针移动
        while i <= j:
            if abs(nums[i]) > abs(nums[j]):
                res[p] = nums[i] * nums[i]
                i += 1
            else:
                res[p] = nums[j] * nums[j]
                j -= 1
                
            p -= 1
            
        return res