数组题一般可以用以下方法尝试解决:
- 如果数组内的数据都是排好序的,那可以用二分查找或者双指针的方法。
- 如果是求连续子区间的问题,有双指针和前缀和两种方法。双指针是通过left和right指针的移动实现,具体做法看下面的题解总结。如果输入数组里都是正数或者都是负数,那么双指针和前缀和都可以求解,双指针的方法时间复杂度会更低。但是如果数组数据的符号不完全一致,那双指针就不起作用了,因为指针的移动无法确定区间内sum的变大还是变小。
- 前缀和是记录前n项的和或者积,再通过求出类似sum[i]-sum[j-1]=k来求证[j, i]区间的和是否符合条件,从而确认j的值。在有正有负的输入数组里使用。
1. 排序数组中两个数字之和
方法一:二分查找
对numbers做低到高遍历,因为已经升序排序,所以可以对每次遍历数的右侧做二分查找。但是这个方法的时间复杂度也不算小,为O(nlogn)。
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 = (low + high) / 2;
if(numbers[i]+numbers[mid] == target){
return new int[]{
i, mid};
} else if(numbers[i]+numbers[mid] < target){
low = mid + 1;
} else{
high = mid -1;
}
}
}
return new int[]{
-1, -1};
}
方法二:双指针
使用一个指针指向头,另一个指向尾。如果和小于target,那么左指针右移;如果和大于target,那么右指针左移;如果和等于target,那么记录下标输出结果。
这种方法不会漏掉结果,因为数组本身已经是升序,并且有且只有唯一解。假设正确的解的下标分别为i, j,那么初始状态下,左指针 <= i < j <=右指针,两个指针逐渐向正解逼近。当其中一个指针先到了正确结果后,另一个指针可能还没到达,此时一定会移动另一个指针来逼近结果。
如果左指针先到达下标 i 的位置,此时右指针还在下标 j 的右侧,sum>target,因此一定是右指针左移,左指针不可能移到 i 的右侧。
如果右指针先到达下标 j 的位置,此时左指针还在下标 ii 的左侧,sum<target,因此一定是左指针右移,右指针不可能移到 j 的左侧。
这个方法比上一个方法的时间复杂度更小,为O(n)。
public int[] twoSum(int[] numbers, int target) {
int i = 0, j = numbers.length-1;
int[] ret = new int[2];
boolean isFound = false;
while(!isFound){
if(numbers[i]+numbers[j] == target){
ret[0] = i;
ret[1