力扣刷题：三数之和（java实现）

最新推荐文章于 2022-02-24 15:39:46 发布

原创最新推荐文章于 2022-02-24 15:39:46 发布 · 262 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法

数据结构专栏收录该内容

27 篇文章

订阅专栏

该博客介绍了如何通过排序和双指针技术高效解决寻找数组中和为0的三元组的问题。避免了三层循环的高时间复杂度，优化算法至线性时间复杂度。具体实现包括对数组排序，然后固定一个元素，用两个指针分别从两侧向中间遍历，找到符合条件的三元组。示例代码展示了这一过程。

题目：给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有和为 0 且不重复的三元组。

标签：数组，双指针，排序

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

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

示例 2：

输入：nums = []
输出：[]

示例 3：

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

提示：

0 <= nums.length <= 3000
10^5 <= nums[i] <= 10^5

题目要求找出和为0的三个数字，如果直接使用三个for循环，会导致时间复杂度很高，比如3000的三次方，时间复杂度有10^9，容易超时。在对数组进行排序的基础上，这时候可以考虑先固定一个数字，那么另外两个数字就是这个数字的负数，另外两个数字要比固定的这个数大，一个从左边开始，一共从右边开始，遍历整个数组，找出符合条件的数。请添加图片描述
大概思路如上图所示，用for循环对固定的那个遍历，用while循环另外两个指针。代码如下：

public List<List<Integer>> threeSum(int[] nums) {
        Arrays.sort(nums);
        List<List<Integer>> res = new ArrayList<>();
        for (int i = 0; i < nums.length-2; i++) {
            //重复的数字跳过
            if(i>0 &&nums[i] == nums[i-1]){
                continue;
            }
            //如果第一个数都大于0了，那后面的都大于0，肯定不符合条件
            if(nums[i] > 0){
                break;
            }
            int left = i+1;//左指针
            int right = nums.length-1;//右指针
            int target = -nums[i];
            while(left<right){
                int sum = nums[left] + nums[right];
                if(sum == target){
                    res.add(Arrays.asList(nums[i],nums[left],nums[right]));
                    while(left<right && nums[left]==nums[left+1]){
                        left++;
                    }
                    while (left<right && nums[right] == nums[right-1]){
                        right--;
                    }
                    left++;
                    right--;
                }else if(sum<target){
                    left++;
                }else {
                    right--;
                }

            }
        }
        return res;

    }

说明一下Arrays.asList() ：
该方法是将数组转化成List集合的方法。

List<String> list = Arrays.asList("a","b","c");