LeetCode19 四数之和
思路 双指针法
和三数之和思路相似
只不过四数之和需要再多加一层for循环以及更加精细的剪枝操作
首先借助库函数Arrarys.sort(nums)
进行快速排序
定义四个指针,i和j分别用于第一层和第二层循环
left和right指针用于定位
如果sum > target
则 right--
如果 sum < target
则 left++
如果 sum = target
则 left和right同时向中间靠拢一位
剪枝操作(去重)非常容易出错,本人调试断点调试近两个小时(难受)
多试多练是最重要的
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;
}
}