Given an array S of n integers, are there elements a, b, c, and d in S such that a + b + c + d = target? Find all unique quadruplets in the array which gives the sum of target.
Note:
Elements in a quadruplet (a,b,c,d) must be in non-descending order. (ie, a ≤ b ≤ c ≤ d)
The solution set must not contain duplicate quadruplets.
Solution 1:递归解法超时,忘记了数组已经排好序,最里层2sum的时候直接用两个指针即可
public class Solution {
public List<List<Integer>> fourSum(int[] num, int target) {
Arrays.sort(num);
List<List<Integer>> res = sub(num, target, 0, 4);
return res == null ? new ArrayList<List<Integer>>() : res;
}
private List<List<Integer>> sub(int[] num, int target, int index, int numbers) {
List<List<Integer>> result = new ArrayList<List<Integer>>();
if (target == 0 && numbers == 0) {
result.add(new ArrayList<Integer>());
return result;
}
if (numbers < 0 || (index >= num.length || num[index] > target)) {
return null;
}
for (int i = index; i < num.length; i++) {
List<List<Integer>> lists = sub(num, target - num[i], i+1, numbers-1);
if (lists != null && lists.size() > 0) {
for (List<Integer> list : lists) {
list.add(0, num[i]);
result.add(list);
}
}
int tmp = i + 1;
while (tmp < num.length && num[tmp] == num[i]) {
tmp++;
}
i = tmp - 1;
}
return result;
}
}
Solution 2:
注意最里层对于重复数字的处理,只有当num[start] + num[end] == twoSumTarget满足时,才跳过剩下的重复数字,否则e.g. target= 4, 对于[2,2,4],如果先跳过重复数字的话[2,2]会miss
public class Solution {
public List<List<Integer>> fourSum(int[] num, int target) {
Arrays.sort(num);
List<List<Integer>> result = new ArrayList<List<Integer>>();
for (int i = 0; i < num.length; i++) {
for (int j = i + 1; j < num.length; j++) {
int twoSumTarget = target - num[i] - num[j];
int start = j + 1, end = num.length - 1;
while (start < end) {
if (num[start] + num[end] == twoSumTarget) {
List<Integer> list = new ArrayList<Integer>();
list.add(num[i]);
list.add(num[j]);
list.add(num[start]);
list.add(num[end]);
result.add(list);
int tmp = start + 1;
while (tmp < end && num[tmp] == num[start]) {
tmp++;
}
start = tmp;
tmp = end - 1;
while (tmp > start && num[tmp] == num[end]) {
tmp--;
}
end = tmp;
} else if (num[start] + num[end] < twoSumTarget) {
start++;
} else {
end--;
}
}
int tmp = j + 1;
while (tmp < num.length && num[tmp] == num[j]) {
tmp++;
}
j = tmp - 1;
}
int tmp = i + 1;
while (tmp < num.length && num[tmp] == num[i]) {
tmp++;
}
i = tmp - 1;
}
return result;
}
}
Solution 3:另外一种解法是利用3sum
public class Solution {
public List<List<Integer>> fourSum(int[] num, int target) {
Arrays.sort(num);
List<List<Integer>> result = new ArrayList<List<Integer>>();
for (int i = 0 ;i < num.length - 3; i++) {
List<List<Integer>> tmp = threeSum(num, i + 1, target - num[i]);
if (tmp != null && !tmp.isEmpty()) {
for (List<Integer> list : tmp) {
list.add(0, num[i]);
result.add(list);
}
}
while (i + 1 < num.length && num[i+1] == num[i]) {
i++;
}
}
return result;
}
private List<List<Integer>> threeSum(int[] num, int index, int target) {
List<List<Integer>> result = new ArrayList<List<Integer>>();
for (int i = index; i < num.length - 2; i++) {
int sum = target - num[i];
int start = i+1, end = num.length - 1;
while (start < end) {
if (num[start] + num[end] == sum) {
List<Integer> list = new ArrayList<Integer>();
list.add(num[i]);
list.add(num[start]);
list.add(num[end]);
result.add(list);
while (start + 1 < end && num[start+1] == num[start]) {
start++;
}
while (end - 1 > start && num[end-1] == num[end]) {
end--;
}
start++;
end--;
} else if (num[start] + num[end] < sum) {
while (start + 1 < end && num[start+1] == num[start]) {
start++;
}
start++;
} else {
while (end - 1 > start && num[end-1] == num[end]) {
end--;
}
end--;
}
}
while (i + 1 < num.length && num[i+1] == num[i]) {
i++;
}
}
return result;
}
}