[哈希表]三数之和

Alson_Code

已于 2022-03-25 22:29:02 修改

阅读量259

点赞数

分类专栏：数据结构和算法文章标签：散列表 leetcode 数据结构哈希算法

于 2022-03-23 22:20:42 首次发布

本文链接：https://blog.csdn.net/weixin_42415173/article/details/123698155

版权

数据结构和算法专栏收录该内容

43 篇文章 0 订阅

订阅专栏

一、题目描述

原文链接：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
-10^5 <= nums[i] <= 10^5

二、思路分析

刚刚看到这道题目的时候，最开始想法就是两数之和的做法，利用先求出a+b所有情况并存储起来，然后在求0 - (a+b);
但是麻烦的是题目要求是不能重复的三元组，求0 - (a+b)还得判断去重！去重目前还不是很拿手！那我们再换一种方法呗！

其实可以用双指针的做法！

第一步：定义一个result来存储结果，List<List<Integer>> result = new ArrayList<>();，在对数组进行排序，指针会用到！Arrays.sort(nums);
第二步：遍历数组，索引下标定义为i，定义一个left指针，int left = i + 1;，定义一个right指针，int right = nums.length - 1;
第三步：来一个while (right > left)循环，求出三个指针对应的元素和sum，int sum = nums[i] + nums[left] + nums[right];，如果sum > 0，则right–，如果sum < 0，则left ++，如果相等，则存储元素并把right++;left--

这样结果就出来啦，但是需要注意以下几点：

如果遍历的元素大于0，则直接返回就可以啦，因为排序来，所以后面都是大于0的元素，还怎么可能三个元素相加等于0！
因为不要重复的元素，所以当前一个值和当前值相等的时候，直接下次循环就可以啦！if (i > 0 && nums[i - 1] == nums[i]) continue;

三、AC代码

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> result = new ArrayList<>();
        // 对数组进行排序
        Arrays.sort(nums);

        // 遍历数组
        for (int i = 0; i < nums.length; i++){
            // 遍历的第一个元素大于0，则直接返回
            if (nums[i] > 0) return result;

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

            int left = i + 1;
            int right = nums.length - 1;
            while (right > left){
                int sum = nums[i] + nums[left] + nums[right];
                if (sum > 0){
                    right--;
                }else if (sum < 0){
                    left++;
                }else{
                    result.add(Arrays.asList(nums[i], nums[left], nums[right]));
					// 防止重复元素
                    while(right > left && nums[right - 1] == nums[right]) right--;
                    while(right > left && nums[left + 1] == nums[left]) left++;

                    right--;
                    left++;
                }
            }

        }

        return result;
    }
}