本文基于代码随想录进行自己的理解和总结
有序数组的平方(双指针)
力扣题目:977
给你一个按 非递减顺序 排序的整数数组 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.考虑到数组中可能存在复数,因此需要先将数组中的负数转化为正数,然后再求平方。而在一个有序数组里面,元素平方的最大值只能在数组的两端。因此我们可以使用双指针法,分别指向数组的左右两端,然后将指针指向的元素平方加入结果数组,并将指针指向的元素更新为下一个元素。直到左右指针相遇,则遍历结束。
2.时间复杂度为O(n),空间复杂度为O(n)。
代码解释:
1.定义sorted方法接收一个整数列表然后返回一个新的整数列表
2.设置l , r 分别为左右指针,i是结果数组的索引指针,res是存储结果的数组初始时被填充为float('inf'),即正无穷大,这只是一个占位值,后续会被平方后的值替换。
-5 | 0 | 1 | 2 | 3 | 7 |