给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
思路:先排序一遍,Map存数字和坐标,set去重。
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
Set<List<Integer>> set=new HashSet<List<Integer>>();
Map<Integer,Integer> map=new HashMap<Integer, Integer>();
Arrays.sort(nums);
for(int i=0;i<nums.length;++i){
map.put(nums[i],i);
}
for(int i=0;i<nums.length;++i){
for(int j=i+1;j<nums.length;++j){
int t=-(nums[i]+nums[j]);
if(map.containsKey(t)&&map.get(t)>j){
set.add(new ArrayList<Integer>(Arrays.asList(nums[i],nums[j],t)));
}
}
}
List<List<Integer>> list=new ArrayList<List<Integer>>(set);
return list;
}
}