15. 三数之和

Posted 2021-12-14 disandafeier

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。

注意：答案中不可以包含重复的三元组。

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
[-1, 0, 1],
[-1, -1, 2]]

（1）这道题暴力是O(n^3)，肯定感觉肯定会蹦的。
（2）那么可以先固定两个数字，第三个元素可以用二分搜索来查找，这样就把时间复杂度下降到O(nlgn)了，但是我写到最后发现结果：

执行用时 : 186 ms, 在3Sum的Java提交中击败了18.92% 的用户
内存消耗 : 48.6 MB, 在3Sum的Java提交中击败了88.18% 的用户

哭了，直接上代码：

    public Integer binarySearch(int[] a, int begin, int end, int target)
        Integer i = null;

        while(begin <= end)
            if(begin == end)
                if(a[begin] == target)
                    i = a[begin];
                
                return i;
            

            int mid = (begin + end) / 2;
            if(a[mid] == target)
                i = a[mid];
                return i;
            
            if(a[mid] < target)
                begin = mid + 1;
            else
                end = mid - 1;
            
        
        return i;
    
    public List<List<Integer>> threeSum(int[] nums)

        List<List<Integer>> endList = new ArrayList<>();
        Integer lasti = null;
        Integer lastj = null;
        Integer lastResult = null;
        Arrays.sort(nums);
        int[] a = nums;
        for(int i = 0; i < a.length-2; i++)
            if(lasti != null && a[i] == lasti)
                continue;
            
            for(int j= i+1; j < a.length-1; j++)
                if(lastj != null && a[j] == lastj)
                    continue;
                
                Integer result = binarySearch(a, j+1, a.length-1, -a[i]-a[j]);
                if(result != null)
                    List<Integer> list = new ArrayList<>();
                    list.add(a[i]);
                    list.add(a[j]);
                    list.add(result);
                    endList.add(list);
                
                lastj = a[j];
            
            lasti = a[i];
            lastj = null;
        
        return endList;