先看一眼题目
我的思路:寻找两个数b和c使得b+c==-a
用set是因为set不能存放同样的值的list,最后再把set转换成list返回
public class Solution15 {
public List<List<Integer>> threeSum(int[] nums) {
Set<List<Integer>> set = new HashSet<>();
int a,b,c;
for(int i = 0 ; i < nums.length-2 ; i++){
a=(-1)*nums[i];
for(int j = i + 1 ; j < nums.length-1 ; j++ ){
b = nums[j];
for(int k = j + 1 ; k < nums.length ; k++){
c = nums[k];
if(a == b+c){
int[] arr = new int[]{nums[i],b,c};
Arrays.sort(arr);
List<Integer> list = new ArrayList<>();
list.add(arr[0]);list.add(arr[1]);list.add(arr[2]);
set.add(list);
}
}
}
}
return new ArrayList<>(set);
}
}
提交,超出时间限制!
查看了一下,倒数第三个案例的时候超出时间限制了,这个测试用例数组特别长,因此猜测没通过的原因是因为时间复杂度太高了。
于是想出改进思路:
原来是O(n3)级别的时间复杂度,我把最内层的循环遍历改成了二分法查找,即找c的时候我不是从j+2到nums.length的顺序的一个一个找,而是对这个区间进行二分查找,时间复杂度降低到O(n2logn),再次提交,就通过了。
debug注意点:二分查找的结束位置要好好思考一下。
代码:
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
Set<List<Integer>> set = new HashSet<>();
int a, b, c;
for (int i = 0; i < nums.length - 2; i++) {
a = (-1) * nums[i];
for (int j = i + 1; j < nums.length - 1; j++) {
b = nums[j];
int u = j + 1, v = nums.length - 1;
for (int k; u <= v; ) {
k = (u + v + 1 ) / 2;
c = nums[k];
if (a == b + c) {
List<Integer> list = new ArrayList<>();
list.add(nums[i]);
list.add(nums[j]);
list.add(nums[k]);
set.add(list);
break;
}else if(k==v)v--;
else if (b + c < a) {
u = k;
} else {
v = k;
}
}
}
}
return new ArrayList<>(set);
}
}
总结:提交以后仅打败5%的人,说明时间复杂度还是太高了,肯定还有更好的办法。不过反正提交成功了,就先不看更好的答案了。