https://leetcode.com/problems/3sum/
这是一个非常经典的题目,各种算法教材上都有示例,所以题意解析就忽略了。
解题思路:
暴力方法肯定是3次方
所以要先排序,然后用第一题2Sum的方法,算法复杂度就变为二次方
但发现仍然是超时,这时就应该想到,需要剪枝,剪掉一些冗余的情况。。。。
思考一下,冗余的情况有两大类,第一类如下
-3,-2,-1,0,1,2,3
当外层循环到0的时候就可以停止了
第二类是
-3,-3,-2,-1,0,1,2,3
最外层循环中,两个连续的-3只需要处理一个就行了
剪掉这两种情况,就可以AC了
题解代码如下:
<span style="font-size:24px;">public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> result=new ArrayList<List<Integer>>();
Arrays.sort(nums);
for(int i=0;i<nums.length-2;i++){
if(nums[i]>0){
break;
}
if(nums[i]==0 && i+2<nums.length){
if(nums[i+2]==0){
List<Integer> triplet=new ArrayList<Integer>();
Collections.addAll(triplet,0,0,0);
result.add(triplet);
}
break;
}
if(i>0 && nums[i]==nums[i-1]){
continue;
}
int twoSum=0-nums[i];
int j=i+1,k=nums.length-1;
while(j<k){
if(nums[j]+nums[k]>twoSum){
k--;
}else if(nums[j]+nums[k]<twoSum){
j++;
}else{
List<Integer> triplet=new ArrayList<Integer>();
Collections.addAll(triplet,nums[i],nums[j],nums[k]);
if(!result.contains(triplet)){
result.add(triplet);
}
j++;
k--;
}
}
}
return result;
}</span>