977. 有序数组的平方

链接

给定一个按非递减顺序排序的整数数组 A，返回每个数字的平方组成的新数组，要求也按非递减顺序排序。

示例 1:

输入：[-4,-1,0,3,10]
输出：[0,1,9,16,100]

示例 2:

输入：[-7,-3,2,3,11]
输出：[4,9,9,49,121]

提示

• 1 <= A.length <= 10000
• -10000 <= A[i] <= 10000
• A 已按非递减顺序排序。

golang语言版

// 解法1
// func sortedSquares(A []int) []int {
// 	arr := make([]int, 0)
// 	for _, v := range A {
// 		v *= v
// 		arr = append(arr, v)
// 	}
// 	sort.Ints(arr)
	
//     return arr
// }

// 解法2
func sortedSquares(A []int) []int {
	size := len(A)
	res := make([]int, size)
	for l, r, i := 0, size-1, size-1; l <= r; i-- {
		if A[l]+A[r] < 0 {
			res[i] = A[l] * A[l]
			l++
		} else {
			res[i] = A[r] * A[r]
			r--
		}
	}
	return res
}