两数之和 2–输入有序数组
讨论 1
二分查找
在数组中找到两个数,使得它们的和等于目标值,可以首先固定第一个数,然后寻找第二个数,第二个数等于目标值减去第一个数的差。利用数组的有序性质,可以通过二分查找的方法寻找第二个数。为了避免重复寻找,在寻找第二个数时,只在第一个数的右侧寻找。
class Solution {
public int[] twoSum(int[] numbers, int target) {
for (int i = 0; i < numbers.length; ++i) {
int low = i + 1, high = numbers.length - 1;
while (low <= high) {
int mid = (high - low) / 2 + low;
if (numbers[mid] == target - numbers[i]) {
return new int[]{i + 1, mid + 1};
} else if (numbers[mid] > target - numbers[i]) {
high = mid - 1;
} else {
low = mid + 1;
}
}
}
return new int[]{-1, -1};
}
}
讨论 2
双指针
初始时两个指针分别指向第一个元素位置和最后一个元素的位置。每次计算两个指针指向的两个元素之和,并和目标值比较。如果两个元素之和等于目标值,则发现了唯一解。如果两个元素之和小于目标值,则将左侧指针右移一位。如果两个元素之和大于目标值,则将右侧指针左移一位。移动指针之后,重复上述操作,直到找到答案。
使用双指针的实质是缩小查找范围。那么会不会把可能的解过滤掉?答案是不会。假设numbers[i]+numbers[j]=target 是唯一解,其中 0 ≤ i < j ≤ numbers.length − 1。初始时两个指针分别指向下标 0 和下标 numbers.length−1,左指针指向的下标小于或等于 ii,右指针指向的下标大于或等于 j。除非初始时左指针和右指针已经位于下标 i 和 j,否则一定是左指针先到达下标 i 的位置或者右指针先到达下标 j 的位置。
如果左指针先到达下标 ii 的位置,此时右指针还在下标 j 的右侧,sum>target,因此一定是右指针左移,左指针不可能移到 i 的右侧。
如果右指针先到达下标 jj 的位置,此时左指针还在下标 i 的左侧,sum<target,因此一定是左指针右移,右指针不可能移到 j 的左侧。
由此可见,在整个移动过程中,左指针不可能移到 i 的右侧,右指针不可能移到 j 的左侧,因此不会把可能的解过滤掉。由于题目确保有唯一的答案,因此使用双指针一定可以找到答案。
class Solution {
public int[] twoSum(int[] numbers, int target) {
int low = 0, high = numbers.length - 1;
while (low < high) {
int sum = numbers[low] + numbers[high];
if (sum == target) {
return new int[]{low + 1, high + 1};
} else if (sum < target) {
++low;
} else {
--high;
}
}
return new int[]{-1, -1};
}
}
特殊数组的特征值
讨论 1
排序 + 枚举 + 二分
根据题意并结合 nums[i] 的数据范围为 1e3,我们可以通过「枚举」的方式找到 x,而对于每个 x 的合法性检查,我们需要快速知道 nums 中比 x 大的数的个数,这可以通过「排序 + 二分」来做。
class Solution {
public int specialArray(int[] nums) {
Arrays.sort(nums);
int n = nums.length;
for (int x = 0; x < 1010; x++) {
int l = 0, r = n - 1;
while (l < r) {
int mid = l + r >> 1;
if (nums[mid] >= x) r = mid;
else l = mid + 1;
}
if (nums[r] >= x && x == n - r) return x;
}
return -1;
}
}
讨论 2
排序 + 二分
若不利用 nums[i]nums[i] 的值域偏小(即值域放大到 1e9),我们可以通过「排序 + 两次二分」的操作来做:
- 第一次二分找分割点 x(二分范围为 [0, 1e9][0,1e9]);
- 第二次则是在 getCnt 函数内部中使用二分来对 x 进行合法性检查。
我们知道若真实的特殊值为 x,那么在以 x 为分割点的数轴上具有「二段性」,假设当前值为 k:
小于真实特殊值 x 的 k 值满足「nums 中大于等于 k 的元素个数超过 k 个」
大于等于真实特殊值 x 的 k 值满足「nums 中大于等于 k 的元素个数不超过 k 个」
因此可以通过「二分」来找真实特殊值为 x,至于 x 的合法检查则和解法一相同,先通过排序确保数组有序,再通过二分的方式来统计大于等于 x 的数的个数。
class Solution {
int[] nums;
public int specialArray(int[] _nums) {
nums = _nums;
Arrays.sort(nums);
int l = 0, r = (int) 1e9;
while (l < r) {
int mid = l + r >> 1;
if (getCnt(mid) <= mid) r = mid;
else l = mid + 1;
}
return getCnt(r) == r ? r : -1;
}
int getCnt(int x) {
int n = nums.length, l = 0, r = n - 1;
while (l < r) {
int mid = l + r >> 1;
if (nums[mid] >= x) r = mid;
else l = mid + 1;
}
return nums[r] >= x ? n - r : 0;
}
}
讨论 3
模拟(计数 + 枚举)
另外一种除非过大缩放 nums 的长度 n,否则仅有代码量优势的做法是使用「计数 + 枚举」的模拟做法。
先使用静态数组对nums 进行词频统计,随后通过逆序枚举的方式找特殊值 x,同时使用 tot 统计遍历过的桶的总元素个数,当满足 tot = x 时,返回结果。
class Solution {
public int specialArray(int[] nums) {
int[] cnts = new int[1010];
for (int x : nums) cnts[x]++;
for (int i = 1009, tot = 0; i >= 0; i--) {
tot += cnts[i];
if (i == tot) return i;
}
return -1;
}
}