1、15. 三数之和
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
方法1:回溯,超时
//用回溯法,超时了~~~
public List<List<Integer>> threeSum1(int[] nums) {
List<List<Integer>> res = new ArrayList<>();
Arrays.sort(nums);
backTrack(nums, res, new ArrayList<Integer>(), 0);
return res;
}
private void backTrack(int[] nums, List<List<Integer>> res, ArrayList<Integer> list, int start) {
if (list.size() == 3) {
if (sumList(list) == 0) {
Collections.sort(list);
res.add(new ArrayList<>(list));
} else {
return;
}
} else {
for (int i = start; i < nums.length; i++) {
//注意判断这:不然有两个一样的{-1,0,1}
if (i == start || nums[i] != nums[i - 1]) {
list.add(nums[i]);
backTrack(nums, res, list, i + 1);
list.remove(list.size() - 1);
}
}
}
}
private int sumList(ArrayList<Integer> list) {
return list.get(0) + list.get(1) + list.get(2);
}
方法2: 思路是先对数组排序,然后固定一个数(遍历),求得结果
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> res = new ArrayList<>();
Arrays.sort(nums);
for (int i = 0; i < nums.length - 2; i++) {
if (nums[i] > 0)
return res;
while (i > 0 && i < nums.length && nums[i] == nums[i - 1])
i++;
int j = i + 1;
int k = nums.length - 1;
while (j < k) {
int sum = nums[i] + nums[j] + nums[k];
if (sum == 0) {
res.add(Arrays.asList(nums[i], nums[j], nums[k]));
//注意这里比较 j < k 就行了
while (++j < k && nums[j] == nums[j - 1]) ;
//j < --k,先--k,再比较 j 和 k 的大小
while (j < --k && nums[k] == nums[k + 1]) ;
} else if (sum < 0) {
while (++j < k && nums[j] == nums[j - 1]) ;
} else {
while (j < --k && nums[k] == nums[k + 1]) ;
}
}
}
return res;
}
2、16. 最接近的三数之和
给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。
例如,给定数组 nums = [-1,2,1,-4], 和 target = 1.
与 target 最接近的三个数的和为 2. (-1 + 2 + 1 = 2).
public int threeSumClosest(int[] nums, int target) {
Arrays.sort(nums);
int closest = nums[0] + nums[1] + nums[2];
for (int i = 0; i < nums.length - 2; i++) {
int j = i + 1;
int k = nums.length - 1;
while (j < k) {
int sum = nums[i] + nums[j] + nums[k];
if (sum == target) {
return sum;
} else if (sum < target) {
while (++j < k && nums[i] + nums[j] + nums[k] < target) ;
if (Math.abs(closest - target) > Math.abs(nums[i] + nums[j - 1] + nums[k] - target))
closest = nums[i] + nums[j - 1] + nums[k];
} else {
while (j < --k && nums[i] + nums[j] + nums[k] > target) ;
if (Math.abs(closest - target) > Math.abs(nums[i] + nums[j] + nums[k + 1] - target))
closest = nums[i] + nums[j] + nums[k + 1];
}
}
}
return closest;
}