Java算法 每日一题 编号15：三数之和

Posted 2022-12-04 烟锁迷城

本系列为每日算法，主要借助@代码随想录与LeetCode网站的解析，仅做记录与鞭策使用。

编号15：三数之和

给你一个包含 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
-105 <= nums[i] <= 105

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解析：

暴力破解的思路还是要对整个数组进行三重嵌套循环，但是双指针法已经为我们揭示了一个可以减少循环的方式，因此这次我们需要两个指针来进行循环的减少。

但是，要记得先排序。

public static List<List<Integer>> threeSum(int[] nums) 
    //进行一次排序，将数组排序
    Arrays.sort(nums);
    //返回结果
    List<List<Integer>> result = new ArrayList<>();
    //循环，取出定位的元素
    for (int i = 0; i < nums.length; i++) 
        //排序后，如果首个元素大于0，就不用继续向后查询了
        if (nums[i] > 0) 
            continue;
        
        //重复的定位元素可以不用再寻找
        if (i > 0 && nums[i] == nums[i - 1]) 
            continue;
        
        //首位，i+1
        int left = i + 1;
        //末位，数组最后一位
        int right = nums.length - 1;
        //当左侧位小于右侧位时继续循环
        while (left < right) 
            //获取三数之和
            int sum = nums[i] + nums[left] + nums[right];
            //如果数据小于预期，则左侧向前
            if (sum < 0) 
                left++;
            //如果数据大于预期，则右侧向后
             else if (sum > 0) 
                right--;
            //如果相等，证明符合预期
             else 
                List<Integer> list = new ArrayList<>();
                list.add(nums[i]);
                list.add(nums[left]);
                list.add(nums[right]);
                result.add(list);
                //得到预期数字之后，开始去重
                //如果符合循环，且下一个预期元素与当前元素相同，就要被除去
                while (left < right && nums[left] == nums[left + 1]) 
                    left++;
                
                //如果符合循环，且下一个预期元素与当前元素相同，就要被除去
                while (left < right && nums[right] == nums[right - 1]) 
                    right--;
                
                //符合预期之后，右侧左侧的数据同时符合，需要同时移除
                right--;
                left++;
            
        
    
    return result;