LeetCode -- 4Sum

Question:

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.

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)

Analysis: 这是2Sum的变种。
1. 暴力求解，4个for循环，依次求和，找到与target相等的元素，但是时间代价高，容易时间超时。
代码如下：

public class Solution {
public List<List<Integer>> fourSum(int[] nums, int target) {
int[] nums1 = sort(nums);
List<List<Integer>> list = new ArrayList<List<Integer>>();

for(int i=0; i<nums1.length; i++) {
for(int j=i+1; j<nums1.length; j++) {
for(int k=j+1; k<nums1.length; k++) {
for(int t=k+1; t<nums1.length; t++) {
int sum = nums1[i] + nums1[j] + nums1[k] + nums1[t];
if(sum == target) {
List<Integer> l = new ArrayList<Integer>();
l.add(nums1[i]);
l.add(nums1[j]);
l.add(nums1[k]);
l.add(nums1[t]);
list.add(l);
}
}
}
}
}

return list;
}

public int[] sort(int[] l) {
for(int i=0; i<l.length; i++) {
for(int j=i+1; j<l.length; j++) {
if(l[i] > l[j]) {
int t = l[i];
l[i] = l[j];
l[j] = t;
}

}
}
return l;
}
}

2. 方法二。首先对数组排序，用两层for循环，然后用两个指针分别指向剩余元素中最大的和最小的，四个数相加后若大于target则大指针前移，若小于target则小指针后移。注意避免重复放入结果集中。

代码如下：

public class Solution {
public List<List<Integer>> fourSum(int[] nums, int target) {
Arrays.sort(nums);

List<List<Integer>> res = new ArrayList<List<Integer>>();
HashSet<List<Integer>> set = new HashSet<List<Integer>>();

for(int i=0; i<nums.length; i++) {
for(int j=i+1; j<nums.length; j++) {
int k = j + 1;
int l = nums.length - 1;
while(k < l) {
int sum = nums[i] + nums[j] + nums[k] + nums[l];
if(sum < target)
k++;
else if(sum > target)
l--;
else {
List<Integer> li = new ArrayList<Integer>();
li.add(nums[i]);
li.add(nums[j]);
li.add(nums[k]);
li.add(nums[l]);
if(!set.contains(li)) {
res.add(li);
set.add(li);
}
k++;
l--;
}
}
}
}
return res;
}
}