题目
给你一个由 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个复合要求的数字,注意第一个for的剪枝
java
import java.util.List;
import java.util.ArrayList;
import java.util.Set;
import java.util.HashSet;
import java.util.Arrays;
public class fourSum {
public static void main(String[] args) {
int[] arr = {1,0,-1,0,-2,2};
List<List<Integer>> res = getFour(arr,0);
for(List<Integer> lis : res) {
for(Integer data : lis) {
System.out.print(data + " ");
}
System.out.println();
}
}
public static List<List<Integer>> getFour(int[] arr,int target) {
int len = arr.length;
Arrays.sort(arr);
Set<List<Integer>> set = new HashSet();
for(int i = 0;i<len-3;i++) {
if(i > 0 && arr[i] == arr[i-1]) {
continue;
}
for(int j = i+1;j<len-2;j++) {
int m = j+1;
int n = len - 1;
while(m<n) {
int data = arr[i]+arr[j]+arr[m]+arr[n];
if(data == target) {
List<Integer> lis = new ArrayList();
lis.add(arr[i]);
lis.add(arr[j]);
lis.add(arr[m]);
lis.add(arr[n]);
set.add(lis);
m++;
n--;
} else if(data < target) {
m++;
} else {
n--;
}
}
}
}
List<List<Integer>> res = new ArrayList();
res.addAll(set);
return res;
}
}