【LeetCode】三数之和双指针 Java 题解

2021 年 12 月 12 日
本文字数：1224 字
阅读完需：约 4 分钟

题目描述

给你一个包含 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]输出：[]
来源：力扣（LeetCode）链接：https://leetcode-cn.com/problems/3sum著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

复制代码

思路分析

今天的算法每日一题是三数之和题目，我们之前做过了两数之和题目，一般是采用 hashmap 来解决的。今天这个题目的解法，采用双指针的思想来解决。那什么是双指针呢？
双指针从广义上来说，是指用两个变量在线性结构上遍历而解决的问题。对于数组，指两个变量在数组上相向移动解决的问题；对于链表，指两个变量在链表上同向移动解决的问题，也称为「快慢指针」问题。
双指针算法通常不难，双指针算法是基于暴力解法的优化。
无论是对于两数之和，还是三数之和，我们需要首先保证数组有序，才能更好的使用这种结构。在无序的输入数组，需要首先排序。
对于三数之和排序去重，

    if (i > 0 && nums[i] == nums[i - 1]) {        continue;    }

复制代码

这个条件很常用，我们应该熟练掌握。

回到三数之和这个问题，我们把他拆解成两数之和问题。具体实现代码如下：

通过代码

class Solution {    public List<List<Integer>> threeSum(int[] nums) {        List<List<Integer>> ans = new ArrayList<>();        Arrays.sort(nums);        int n = nums.length;        for (int i = 0; i < n; i++) {            if (i > 0 && nums[i] == nums[i - 1]) {                continue;            }
            List<List<Integer>> twoSumResult = findTwoSum(nums, i + 1, 0 - nums[i]);            for (List<Integer> temp : twoSumResult) {                ans.add(new ArrayList<Integer>(Arrays.asList(nums[i], temp.get(0), temp.get(1))));            }
        }
        return ans;    }
    public List<List<Integer>> findTwoSum(int[] nums, int start, int target) {        int n = nums.length;        int j = n - 1;        List<List<Integer>> ans = new ArrayList<>();        for (int i = start; i < n; i++){            if (i > start && nums[i] == nums[i - 1]) {                continue;            }            while (i < j && nums[i] + nums[j] > target) {                j--;            }            if (i < j && nums[i] + nums[j] == target) {                ans.add(new ArrayList<Integer>(Arrays.asList(nums[i], nums[j])));            }        }
        return ans;    }}