LeetCode 18.四数之和
/**
* @package: _03hash._18fourSum
* @description:18.四数之和
* @author: Yunyang
* @date: 2024/10/23 14:10
* @version:1.0
**/
public class TestFourSum {
public static void main(String[] args) {
Solution solution = new Solution();
int[] nums1 = {1, 0, -1, 0, -2, 2};
int target1 = 0;
System.out.println(solution.fourSum(nums1, target1)); // 输出: [[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]
int[] nums2 = {2, 2, 2, 2, 2};
int target2 = 8;
System.out.println(solution.fourSum(nums2, target2)); // 输出: [[2,2,2,2]]
int[] nums3 = {1000000000,1000000000,1000000000,1000000000};
int target3= -294967296;
System.out.println(solution.fourSum(nums3, target3));
}
}
class Solution {
public List<List<Integer>> fourSum(int[] nums, int target) {
// 初始化result数组
List<List<Integer>> result = new ArrayList<>();
if(nums == null || nums.length < 4){
return result;
}
// 对数组 nums 进行排序
Arrays.sort(nums);
// 使用两层循环来固定前两个数 nums[i] 和 nums[j],
// 然后在剩下的数组中使用双指针 left 和 right 来查找另外两个数
int n = nums.length;
for (int i = 0; i < n; i++) {
// 提前终止条件
if (nums[i] > target && nums[i] >= 0){
break;
}
// 跳过重复的元素
if(i > 0 && nums[i] == nums[i - 1]){
continue;
}
for (int j = i + 1; j < n - 1; j++) {
// 提前终止条件
if(nums[i] + nums[i] > target && nums[i] + nums[j] >= 0){
break;
}
// 跳过重复的元素
if(j > i + 1 && nums[j] == nums[j - 1]){
continue;
}
int left = j + 1;
int right = n - 1;
while (left < right){
long sum = nums[i] + nums[j] + nums[left] + nums[right];
if (sum == Integer.MIN_VALUE){
return result;
}
if(sum > target){
right--;
} else if(sum < target){
left++;
} else {
result.add(Arrays.asList(nums[i],nums[j],nums[left],nums[right]));
// 跳过重复的元素
while (left < right && nums[left] == nums[left + 1]){
left++;
}
// 跳过重复的元素
while (left < right && nums[right] == nums[right - 1]){
right--;
}
left++;
right--;
}
}
}
}
return result;
}
}