LeetCode 3Sum

54 篇文章 0 订阅

原题链接在这里:https://leetcode.com/problems/3sum/

Two Sum 是这道题的子问题,但是对于3Sum来说又不方便用HashMap,因为不方便存入相同的key,e.g -1,-1,2.

因此,这道题需要先进行排序,对于0 到 length-3 的每一个数后面生成两个指针,一个指向当前元素的下一元素,一个指向最后一个元素,判断三个数的和是否是target,若比target小,头一个指针后移,若比target大,后一个指针前移。

Note: 1. for loop 是注意时候会有溢出,因为会用到至少后面两个数,所以终止条件应该是 i<nums.length - 2;

2. 遇到相同元素时,直接跳过,否者会把相同的 list 存两遍,注意这里是检查是否与上一个元素相同,所以应该比较 nums[i] == nums[i-1],在比较这个之前要加上 i>0,避免overflow.

3. 最后一个else 中记得加入 j++;k--; 否则就是infinite loop, 并且注意j++;k--时是否会遇到相同元素,遇到需跳过。否则会加 入重复的结果,e.g. -2, 0, 0, 2 会加两个 -2,0,2 的 subList.



AC  Code:

public class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> res = new ArrayList<List<Integer>>();
        
        if(nums == null || nums.length < 3)
            return res;
            
        Arrays.sort(nums);
        for(int i = 0 ; i < nums.length-2; i++){ //error
            
            if(i>0 && nums[i] == nums[i-1]){ //error
                continue;
            }
            
            int j = i+1;
            int k = nums.length - 1;
            while(j<k){
                if(nums[i] + nums[j] + nums[k] < 0){
                    j++;
                }else if(nums[i] + nums[j] + nums[k] > 0){
                    k--;
                }else{
                    List<Integer> subList = new ArrayList<Integer>();
                    subList.add(nums[i]);
                    subList.add(nums[j]);
                    subList.add(nums[k]);
                    res.add(subList);
                    
                    j++;
                    k--;
                    
                    while(j<k && nums[j] == nums[j-1]){ //error
                        j++;
                    }
                    while(j<k && nums[k] == nums[k+1]){
                        k--;
                    }
                    
                }
            }
            
        }
        
        return res;
    }
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值