题目
给定一个按非递减顺序排序的整数数组 A,返回每个数字的平方组成的新数组,要求也按非递减顺序排序。
法一:暴力
暴力如此好解
public int[] sortedSquares(int[] A) {
int[] ans = new int[A.length];
for (int i = 0; i < A.length; i++) {
ans[i] = A[i] * A[i];
}
Arrays.sort(ans);
return ans;
}
法二:双指针版本一
设置一个新数组,用于返回结果。
我们可以找到一个临界值,临界值左边是负数,右边是正数。(临界值不一定是0哦,我一开始以为是0,但是一个数组可能就没有0)
记录这个临界值的下标,再设置两个变量,一个left,一个right,left指向左边的元素,right指向右边的元素,我们循环进行比较,如果:
- A[left] * A[left] > A[right] * A[right],则将小的值先放入,right++,再用后面的数与left指向的做比较;
- 不满足条件一,则放入left指向的元素平方,再left–;继续循环
- 但是我们要判断left和right的范围,如果left已经超出范围,但还能进入循环,则说明right还在范围内,将right及其后面的元素的平方加入数组;
- 如果right超过范围了,还能进入循环,则说明left还在范围内,继续将left及其前面的元素的平方加入数组。
应该先去判断条件3和条件4,然后再进行条件一、条件二的判断。
public int[] sortedSquares2(int[] A) {
int index = 0;
int[] ans = new int[A.length];
int k = 0;
// 考虑第一个元素,如果第一个元素是大于0的,就不需要继续找了,该数组都是正数
// 如果第一个元素小于0,我们就要去找分界线,只有存在 >0的某元素,直接break
for (int i = 0; i < A.length; i++) {
if (A[i] < 0) {
index = i;
} else {
break;
}
}
int left = index;
int right = index + 1;
while (left >= 0 || right < A.length) {
if (left < 0) {
ans[k] = A[right] * A[right];
right++;
} else if (right >= A.length) {
ans[k] = A[left] * A[left];
left--;
} else if (A[left] * A[left] > A[right] * A[right]) {
ans[k] = A[right] * A[right];
right++;
} else if (A[left] * A[left] <= A[right] * A[right]) {
ans[k] = A[left] * A[left];
left--;
}
k++;
}
return ans;
}
注意:因为临界值的下标不一定就是在数组的中间位置,要考虑到中间值前面和后面的元素个数不相等的情况,所以while循环条件应该是left >= 0 || right < A.length。
法三:双指针版本二
上面的版本我们是正着来的,我们可以倒着来,这样就避免了讨论越界情况。
public int[] sortedSquares(int[] A) {
int[] ans = new int[A.length];
int index = ans.length - 1;
for (int i = 0, j = A.length - 1; i <= j;) {
if (A[i] * A[i] > A[j] * A[j]) {
ans[index] = A[i] * A[i];
i++;
} else {
ans[index] = A[j] * A[j];
j--;
}
index--;
}
return ans;
}
反思
麻烦作者本人您下次能不能深度思考,不要一没有做出来就烦躁!!
能不能多思考多思考,而且是多冷静多冷静思考!!!!
稳重一点好不好。
end.