题目描述
输入一个递增排序的数组和一个数字S,在数组中查找两个数,使得他们的和正好是S,如果有多对数字的和等于S,则输入任意一对即可。
练习地址-输出有点不同
lc
实现
双指针方法:
- 头尾各设置一个指针
- 根据递增性质,
首尾和=sum,输出两数;
首尾和>sum,尾指针前移
首尾和<sum,头指针后移
时间复杂度 O(N) : N为数组 nums 的长度;双指针共同线性遍历整个数组。
空间复杂度 O(1): 变量 i, j使用常数大小的额外空间。
import java.util.ArrayList;
// 时间O(n)
public class C57_array_SumOfNumber {
public static ArrayList<Integer> findNumbersWithSum(int[] array, int sum) {
ArrayList<Integer> res = new ArrayList<>();
if (array == null || array.length == 0 || sum < 0) {
return res;
}
int p1 = 0;// 头部一个指针
int p2 = array.length - 1;// 尾部一个指针
while (p1 < p2) {
int curSum = array[p1] + array[p2];
if (sum == curSum) {
res.add(array[p1]);
res.add(array[p2]);
break; //♥ 一定要记得加 会死循环
} else if (curSum > sum) {//递增,当前值>15,尾部前移
p2--;
} else {// 递增,当前值<15,头部左移
p1++;
}
}
return res;
}
}
Test
public static void main(String[] args) {
int[] array = {1, 2, 4, 7, 11, 15};
ArrayList<Integer> list = findNumbersWithSum(array, 15);
for (Integer i : list) {
System.out.print(i + " ");
}
}