三数之和-力扣15-java排序+双指针

一、题目描述

给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请

你返回所有和为 0 且不重复的三元组。

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

示例 1：

输入：nums = [-1,0,1,2,-1,-4]

输出：[[-1,-1,2],[-1,0,1]]

解释：

nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。

nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。

nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。

不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。

注意，输出的顺序和三元组的顺序并不重要。

示例 2：

输入：nums = [0,1,1]

输出：[]

解释：唯一可能的三元组和不为 0 。

示例 3：

输入：nums = [0,0,0]

输出：[[0,0,0]]

解释：唯一可能的三元组和为 0 。

来源：力扣（LeetCode）

链接：https://leetcode.cn/problems/3sum

著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

二、运行结果

三、解题思路

首先对原数组按非降序的方式进行排序。逐个将从第1个至第n-2个元素作为三个元素中的第一个（i），另外两个元素只考虑i后面的元素，采用双指针的方式，两个指针分别从两端向中间靠拢，如果出现重复的值就跳过（只考虑第一次出现的时候），避免出现重复的组合。

这里效率比较低的原因可能是在循环里的判断太多，可以尝试一下改进循环内结构，减少判断，如将while循环也改为for循环，即以第二个元素的下标作为内存循环。

四、AC代码

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        int len = nums.length;
        List<List<Integer>> ans = new ArrayList<>();
        Arrays.sort(nums);  //对原数组排序
        for(int i=0; i<len-2; i++){  //第一个元素下标
            if(i>0 && nums[i]==nums[i-1]) continue;
            int j = i+1, k = len -1; //j是第二个元素的下标，k是第三个元素的下标
            while(j < k){
                List<Integer> tmplist= new ArrayList<>();
                if(j > i+1){  //去除重复的组合
                    while(nums[j] == nums[j-1] && j<k) j++;
                }
                if(k < len-1){
                    while(nums[k] == nums[k+1] && k>j) k--;
                }
                if(j >= k) break;
                if(nums[i] + nums[j] + nums[k] == 0){
                    tmplist.add(nums[i]);
                    tmplist.add(nums[j]);
                    tmplist.add(nums[k]);
                    ans.add(tmplist);
                    j++; k--;
                }
                else if(nums[i] + nums[j] + nums[k] > 0) k--;
                else j++;
            }
        }
        return ans;
    }
}