给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> list2 = new ArrayList<List<Integer>>();
// 判断给定数组是否存在三个及以上元素
if(nums.length >= 3){
// 对数组排序并放入HashMap中
Arrays.sort(nums);
Map<Integer,Integer> map = new HashMap<>();
for(int i =0; i<nums.length;i++){
map.put(nums[i],i);
}
List<Integer> list = null;
for(int i=0;i<nums.length;i++){
for(int j=i+1;j<nums.length;j++){
int c = 0-nums[i]-nums[j];
// 判断mapKey值是否存在使三数相加为零,且下标大于j
if(map.containsKey(c) && map.get(c) > j){
list = new ArrayList<Integer>();
list.add(nums[i]);
list.add(nums[j]);
list.add(c);
if(!list2.contains(list)){
list2.add(list);
}
}
}
}
}
return list2;
}