题目描述:
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)
- 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)
思路解析:
- 化繁为简,简化成两个数相加,找目标和的函数
- 注意要先把数组排序,固定一个值,然后low和high分别从i+1和length-1往中间走
- 为了防止重复添加,可以使用HashSet来避免重复
- 判空条件是看num小于3
代码:
import java.util.*;
public class Solution {
public ArrayList<ArrayList<Integer>> threeSum(int[] num) {
ArrayList<ArrayList<Integer>> res = new ArrayList<ArrayList<Integer>>();
if(num == null || num.length<3)
return res;
HashSet<ArrayList<Integer>> hs = new HashSet<ArrayList<Integer>>();
Arrays.sort(num);//千万不要丢排序的这个哦
for(int i=0;i<=num.length-3;i++){
int low = i+1;
int high = num.length-1;
while(low<high){
int sum = num[i]+num[low]+num[high];
if(sum==0){
ArrayList<Integer> util = new ArrayList<Integer>();
util.add(num[i]);
util.add(num[low]);
util.add(num[high]);
if(!hs.contains(util)){
hs.add(util);
res.add(util);
}
low++;
high--;
}else if(sum<0){
low++;
}else{
high--;
}
}
}
return res;
}
}