0977-leetcode算法实现之有序数组的平方sqaure-of-a-sorted-array-python&golang实现
给你一个按 非递减顺序 排序的整数数组 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)
链接:https://leetcode-cn.com/problems/squares-of-a-sorted-array
python
# 有序数组的平方
class Solution:
def sortedSquares(self, nums: [int]) -> [int]:
"""
双指针,时间O(n),空间除存储空间外O(1)
思路:
1.i,j即左右指针,每次移动,逆序将对应大的平方加入新开辟的数组,
2.如果i平方大,对应加入ans,同时i++,否则j平方加入,j--,
3.重复1,2,每次循环pos--
4.最后返回ans
:param nums:
:return:
"""
n = len(nums)
ans = [0] * n
i, j, pos = 0, n-1, n-1
while i <= j:
if nums[i] * nums[i] > nums[j] * nums[j]:
ans[pos] = nums[i] * nums[i]
i += 1
else:
ans[pos] = nums[j] * nums[j]
j -= 1
pos -= 1
return ans
def sortedSqaures(self, nums: [int]):
"""
直接排序,时间O(nlogn),空间O(logn)->排序空间
:param nums:
:return:
"""
return sorted(num * num for num in nums)
if __name__ == "__main__":
nums = [-4,-3,1,4,6,7]
test = Solution()
print(test.sortedSqaures(nums))
golang
//+build ignore
package main
import "fmt"
func main() {
nums1 := []int{-4, -3, 1, 4, 6, 7}
fmt.Println(sortedSqaures(nums1))
}
func sortedSqaures(nums []int) []int {
var n int = len(nums)
var i int = 0
var j int = n - 1
var pos int = n - 1
ans := make([]int, n)
for i <= j { // 根据数组的有序性,每次将值大者逆序赋值
lm, rm := nums[i]*nums[i], nums[j]*nums[j]
if lm > rm {
ans[pos] = lm
i++ // 左指针右移,指向下一个元素
} else {
ans[pos] = rm
j-- // 右指针左移,指向下一个元素
}
pos-- // ans的待填充元素更新
}
return ans
}