给定一个包含 n 个整数的数组 nums 和一个目标值 target,判断 nums 中是否存在四个元素 a,b,c 和 d ,使得 a + b + c + d 的值与 target 相等?找出所有满足条件且不重复的四元组。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/4sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
完全仿照三数之和。自己写的,有点乱。
class Solution {
public List<List<Integer>> fourSum(int[] nums, int target) {
List<List<Integer>> ans = new ArrayList<List<Integer>>();
int n = nums.length ;
Arrays.sort(nums);
for(int first = 0 ; first < n ; first ++){//第一层
if(first > 0 && nums[first] == nums[first - 1]){
continue ;
}
for(int second = first + 1 ; second < n ; second ++){
if(second > first + 1 && nums[second] == nums[second - 1]){
continue ;
}
int forth = n-1 ;
int pretarget = target - nums[first] - nums[second] ;
for(int third = second + 1 ; third < n ; third ++ ){
if(third > second + 1 && nums[third] == nums[third - 1]){
continue ;
}
while(third < forth && nums[third] + nums[forth] > pretarget){
forth--;
}
if(third == forth){
break ;
}
if(nums[third] + nums[forth] == pretarget){
List<Integer> list = new ArrayList <Integer> () ;
list.add(nums[first]);
list.add(nums[second]);
list.add(nums[third]);
list.add(nums[forth]);
ans.add(list);
}
}
}
}
return ans ;
}
}
感觉写的不太好,但是暂时先这样写吧。