有序數(shù)組的平方
給你一個按 非遞減順序 排序的整數(shù)數(shù)組 nums
叮喳,返回 每個數(shù)字的平方 組成的新數(shù)組床佳,要求也按 非遞減順序 排序啤呼。
示例 1:
輸入:nums = [-4,-1,0,3,10]
輸出:[0,1,9,16,100]
解釋:平方后弓叛,數(shù)組變?yōu)?[16,1,0,9,100]
排序后镶苞,數(shù)組變?yōu)?[0,1,9,16,100]
示例 2:
輸入:nums = [-7,-3,2,3,11]
輸出:[4,9,9,49,121]
方法一:直接排序
public int[] sortedSquares(int[] nums) {
int[] res = new int[nums.length];
for (int i = 0; i < nums.length; i++) {
res[i] = nums[i] * nums[i];
}
Arrays.sort(res);
return res;
}
方法二:雙指針
根據(jù)有序特點肝断,先找到最后一個負數(shù)的位置蝶柿,對于平方丈钙,其左邊的結(jié)果從右往左遞增交汤,其右邊的結(jié)果從左往右遞增
public int[] sortedSquares(int[] nums) {
int[] res = new int[nums.length];
int lastNegative = 0;
for (int i = 0; i < nums.length; i++) {
if (nums[i] < 0) {
lastNegative = i;
} else {
break;
}
}
int left = lastNegative, right = lastNegative + 1;
int i = 0;
while (left >= 0 || right < nums.length) {
int mul1 = left >= 0 ? nums[left] * nums[left] : Integer.MAX_VALUE;
int mul2 = right < nums.length ? nums[right] * nums[right] : Integer.MAX_VALUE;
if (mul1 < mul2) {
res[i++] = mul1;
left--;
} else {
res[i++] = mul2;
right++;
}
}
return res;
}
雙指針2
數(shù)組平方從兩邊向中間遞減
public int[] sortedSquares(int[] nums) {
int[] res = new int[nums.length];
int left = 0, right = nums.length - 1;
int i = nums.length - 1;
while (left <= right) {
int mul1 = nums[left] * nums[left];
int mul2 = nums[right] * nums[right];
if (mul1 > mul2) {
res[i--] = mul1;
left++;
} else {
res[i--] = mul2;
right--;
}
}
return res;
}