题目
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: The solution set must not contain duplicate quadruplets.
给一个包含n个数的整数数组S,在S中找到所有使得和为给定整数target的四元组(a, b, c, d)。
样例
For example, given array S = [1, 0, -1, 0, -2, 2], and target = 0.
A solution set is:
[
[-1, 0, 0, 1],
[-2, -1, 1, 2],
[-2, 0, 0, 2]
]
例如,对于给定的整数数组S=[1, 0, -1, 0, -2, 2] 和 target=0. 满足要求的四元组集合为:
(-1, 0, 0, 1)
(-2, -1, 1, 2)
(-2, 0, 0, 2)
解题思路
针对4 Sum这个题,我的思路还是两个指针 Two Pointers,时间复杂度是O(n^3)。
针对这个题,有这么一点是要特别注意的,就是 The solution set must not contain duplicate quadruplets.,我们要确保每一组答案都不包含数组中原先已经存在的元素。
实现过程是,从当前元素i开始,寻找下一个元素i+1,再从i+1之后开始双指针,如果结果小于目标值,start增加;如果结果大于目标值,end减小。当等于目标值的时候,输出结果。
具体代码如下:
public ArrayList<ArrayList<Integer>> fourSum(int[] nums, int target) {
ArrayList<ArrayList<Integer>> res = new ArrayList<ArrayList<Integer>>();
Arrays.sort(nums);
for (int i = 0; i < nums.length; i++) {
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
for (int j = i + 1; j < nums.length; j++) {
if (j > i + 1 && nums[j] == nums[j - 1]) {
continue;
}
int start = j + 1, end = nums.length - 1;
while (start < end) {
int sum = nums[i] + nums[j] + nums[start] + nums[end];
if (sum == target) {
ArrayList<Integer> temp = new ArrayList<Integer>();
temp.add(nums[i]);
temp.add(nums[j]);
temp.add(nums[start]);
temp.add(nums[end]);
res.add(temp);
while (start < end && nums[start] == nums[start + 1]) {
start++;
}
while (start < end && nums[end] == nums[end - 1]) {
end--;
}
start++;
end--;
} else if (sum < target) {
start++;
} else {
end--;
}
}
}
}
return res;
}