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: 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] ]
这道题的意思是:有一组数组,需要我们找到数组中的三个数字能够使这三个数字的和是0,用列表的形式返回这三个数字。首先对数组进行排序,然后可以用sun表示除了第一个数字外其他两个数字的和,然后用二分查找的方法去判断索引为lo的数字和索引为hi的数字的和与sum的大小,如果等于sum,则把这三个数字以列表的形式添加进列表中,并且lo增加1,hi减1,如果这两个数字的和大于sum,这时候因为数组有序,可以通过减小hi使nums[hi]减小,反之则增加lo。不过在整个for循环中要保证数字不重复,但是不能用nums[i]!=nums[i+1]控制,这样程序会直接跳过nums[i]。
List<List<Integer>> res = new ArrayList<List<Integer>>();
Arrays.sort(nums);
for(int i=0;i<nums.length-2;i++){
if(i==0||i>0&&nums[i]!=nums[i-1]){
int lo = i+1;
int hi = nums.length-1;
int sum = 0-nums[i];
//System.out.println(sum);
while(lo<hi){
if(nums[lo]+nums[hi]==sum){
res.add(Arrays.asList(nums[i],nums[lo],nums[hi]));
while(lo<hi&&nums[lo]==nums[lo+1]){lo++;}
while(lo<hi&&nums[hi]==nums[hi-1]){hi--;}
lo++;hi--;
}else if(nums[lo]+nums[hi]>sum){hi--;}
else lo++;
}
}
}
return res;