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)题意:给定一个数组,找出其中的三个数,使之三个数的和为0,并输出这三个数,按从小到大输出,如果有多组就全部输出。
分析:在 LeetCode 解题报告 Two Sum 这篇文章中介绍了三种方法,此题是这个题的升级版,这个题如果用暴力解决那么将是O(n3)的时间复杂度,如果用HashMap也是不方便,因为数据是可重复的。
接下来就直接看代码
public List<List<Integer>> threeSum(int[] num) {
Arrays.sort(num);
List<List<Integer>> list = new ArrayList<List<Integer>> ();
int first, end, mid;
//遍历数组
for(int i = 0; i < num.length-2; i++) {
if(i==0 || num[i] > num[i-1]) {
first = i;
end = num.length - 1;
mid = first + 1;
if(num[first] > 0 || num[end] < 0)
break;
while(mid < end) {
int sum = num[first] + num[mid] + num[end];
if(sum == 0) {
ArrayList<Integer> each = new ArrayList<Integer>();
each.add(num[first]);
each.add(num[mid]);
each.add(num[end]);
if(!list.contains(each))
list.add(each);
mid++;
end--;
while(mid < end && num[end] == num[end-1])
end--;
while(mid < end && num[mid] == num[mid+1])
mid ++;
}
else if(sum < 0) {
mid++;
}
else
end--;
}
}
}
return list;
}
思路其实就很简单了,就是锁定第一个值遍历,剩下的两个的数,按照数组中寻找两个数的和是定值来解决就ok了。
如果按着思路写出的代码提交会有出现超时现象,那么我们就要优化了。
这里面做了很多的优化,比如:
if(i==0 || num[i] > num[i-1])
这里是防止相同数据输入
while(mid < end && num[end] == num[end-1])
end--;
while(mid < end && num[mid] == num[mid+1])
mid ++;
这里避免不必要的重复