给定一个包含 n 个整数的数组 nums
,判断 nums
中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
思路:
两数之和的方法:通过扫描一次同时查询并记录HashMap的方式,能够找到某一个符合条件的组合,但是对于当前这道题,会有重复的结果,所以这种方法不行。
采用双指针的方法:先对数组进行排序,然后外层循环选定第一个数,在选定数后方初始化前后指针;前后指针的值大于target - 选定数,移动后指针;小于,移动前指针;等于,移动后指针(或前指针);直到前后指针重合。
为什么可以用这种双指针的方法:因为首先数组是排序好的,然后考察两指针指向的数的和是否等于target,若大于tar,这时候右移左指针和会更大,更不可能和等于tar,右移右指针同理;左移左指针也不行,因为那种情况已经判断过了。所以只有按照上述移动方法,才有可能出现符合条件的结果。
为了避免重复结果的出现,有两个地方注意:
1、选定第一个数的时候,可以筛除重复的
eg:-1, -1, 0, 0, 0,1 不筛除会出现重复的结果
2、在选定第一个数之后,第二个重复也会出现重复结果(方法是:出现相等的情况就跳过所有重复元素)
eg:0, 0, 0, 0 对第二个数不做重复判断,也会导致重复结果
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> res = new ArrayList<List<Integer>>();
Arrays.sort(nums);
for (int i = 0; i < nums.length - 2; i++) {
if(i != 0 && nums[i] == nums[i - 1])
continue;
int low = i + 1;
int high = nums.length - 1;
while (low < high){
if(nums[low] + nums[high] == -nums[i]){
List<Integer> list = new ArrayList<>();
list.add(nums[i]);
list.add(nums[low]);
list.add(nums[high]);
res.add(list);
do {
high--;
}while (low < high && nums[high + 1] == nums[high]);
}else if(nums[low] + nums[high] > -nums[i])
high--;
else
low++;
}
}
return res;
}