您的位置:首页 > 编程语言 > Java开发

LEETCODE 15 3Sum (JAVA题解)

2016-02-21 22:06 841 查看
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>
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息