给定一个按非递减顺序排序的整数数组 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 已按非递减顺序排序。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/squares-of-a-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
1.将负数取反;
2.利用双指针,在正负交界处比较,小者先加入到结果数组中;
3.剩下的依次加入结果数组中。
代码如下:
/*try second */
public int[] sortedSquares(int[] A) {
int len = A.length;
int i = 0;
for (; i < len; i++) {
if(A[i] < 0){
A[i] = -1 * A[i];
}else {
break;
}
}
int left = i-1;//交界处的左面,之前的负数的开始
int right = i;//交界处的右面,正数的开始
int[] res = new int[len];
int k = 0;//新数组的下标
while (left >= 0 && right < len){
if(A[left] <= A[right]){
res[k++] = A[left] * A[left];
left--;
}else {
res[k++] = A[right] * A[right];
right++;
}
}
while(right < len){
res[k++] = A[right] * A[right];
right++;
}
while(left >= 0){
res[k++] = A[left] * A[left];
left--;
}
return res;
}
/*try first*/
// public int[] sortedSquares(int[] A) {
// int len = A.length;
// for (int i = 0; i < len; i++) {
// if(A[i] < 0){
// A[i] = Math.abs(A[i]);
// }else {
// break;
// }
// }
//
// Arrays.sort(A);
//
// int res[] = new int[len];
// for (int i = 0; i < len; i++) {
// res[i] = A[i] * A[i];
// }
// return res;
// }