LeetCode19 四数之和
给你一个由 n 个整数组成的数组 nums ,和一个目标值 target 。请你找出并返回满足下述全部条件且不重复的四元组
[nums[a], nums[b], nums[c], nums[d]] (若两个四元组元素一一对应,则认为两个四元组重复):
0 <= a, b, c, d < n a、b、c 和 d 互不相同 nums[a] + nums[b] + nums[c] +
nums[d] == target 你可以按 任意顺序 返回答案 。
示例 1:
输入:nums = [1,0,-1,0,-2,2], target = 0
输出:[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]] 示例 2:
输入:nums = [2,2,2,2,2], target = 8 输出:[[2,2,2,2]]
提示:
1 <= nums.length <= 200
-109 <= nums[i] <= 109
-109 <= target <= 109
思路 双指针法
和三数之和思路相似
只不过四数之和需要再多加一层for循环以及更加精细的剪枝操作
首先借助库函数Arrarys.sort(nums)
进行快速排序
定义四个指针,i和j分别用于第一层和第二层循环
left和right指针用于定位
如果sum > target
则 right--
如果 sum < target
则 left++
如果 sum = target
则 left和right同时向中间靠拢一位
剪枝操作(去重)非常容易出错,本人调试断点调试近两个小时(难受)
多试多练是最重要的
java
import com.sun.net.httpserver.Authenticator.Result;
class Solution {
public List<List<Integer>> fourSum(int[] nums, int target) {
Arrays.sort(nums); //快速排序 O(nlogn)
List<List<Integer>> res = new ArrayList<>();
int len = nums.length;
for(int i = 0; i< len - 3; i++){
//跳出重复
if(nums[i] > 0 && nums[i] >= target) return res;
//对nums[i]去重
if(i > 0 && nums[i -1] == nums[i]) continue;
for (int j = i + 1; j < len - 2; j++){
if(j > i + 1 && nums[j] == nums[j - 1]){
continue;
}
int left = j + 1;
int right = len - 1;
while(left < right) {
int sum = nums[i] + nums[j] + nums[left] + nums[right];
if(sum == target){
res.add(Arrays.asList(nums[i], nums[j], nums[left], nums[right]));
// 对nums[left]去重
while(left < right && nums[left+1] == nums[left]) ++left;
对nums[right]去重
while(left < right && nums[right-1] == nums[right]) --right;
//找到答案,双指针同时收缩
++left;
--right;
}else if( sum > target){
--right;
}else if(sum < target){
++left;
}
}
}
}
return res;
}
}