LeetCode 3Sum

来源:互联网 发布:网络存储器怎么用 编辑:程序博客网 时间:2024/06/02 03:26

原题链接在这里: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