Given an array S of n integers, are there elements a, b, c in S such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.
Note:
Elements in a triplet (a,b,c) must be in non-descending order. (ie, a ≤ b ≤ c)
Note:
Elements in a triplet (a,b,c) must be in non-descending order. (ie, a ≤ b ≤ c)
The solution set must not contain duplicate triplets.
For example, given array S = {-1 0 1 2 -1 -4},
A solution set is:
(-1, 0, 1)
(-1, -1, 2)
分析:
Ksum的问题,因为规定了元素数量,不好用回溯做,因此要列举所有情况,过程中,保证3个元素下标的有序就可以了。
需要注意的是,因为S中可能存在重复的数字,所以即便保证了三个下标的有序,还是有可能找到重复的结果,所以需要一个HashSet来判断重复。
public class Solution {
public List<List<Integer>> threeSum(int[] num) {
List<List<Integer>> res = new ArrayList<List<Integer>>();
if(num == null || num.length < 3) return res;
HashSet<List<Integer>> set = new HashSet<List<Integer>>();
Arrays.sort(num);
for(int i=0; i< num.length-2; i++){
int j=i+1;
int k=num.length-1;//利用排序的特点减小复杂度
while(j<k){
int sum = num[i]+num[j]+num[k];
if(sum<0)
j++;
else if(sum>0)
k--;
else{
List<Integer> item = new ArrayList<Integer>();
item.add(num[i]);
item.add(num[j]);
item.add(num[k]);
//因为num可能存在重复元素,所以i,j,k的组合也可能重复
if(!set.contains(item)){
res.add(item);
set.add(item);
}
j++;
k--;
}
}
}
return res;
}
}