【LeetCode】三数之和 Java 题解

关注

发布于: 1 小时前

题目描述

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

复制代码

思路分析

三数之和是数组类型题目，也是两数之和的进阶版。
对于数组类题目，对数组元素进行排序可以简化思路，提升算法效率。
这个题目还有一个难点是，不可以包含重复的三元组。具体对于每一层循环而言，相邻的元素不能相同。
三层循环的实现代码，在提交的时候显示超时，未通过。优化方向需要借鉴两数之和的思想，将 -nums[i] 设置为 target, 然后使用双指针，对排序数组从首尾进行遍历，得出答案。

通过代码

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

复制代码

总结

通过代码的时间复杂度是 O(n * n),快速排序使用了额外的空间，空间复杂度是 O(log n)
这是一个经典的题目，之前做过，但这次没有通过，要多总结，多反思，掌握好这个题目。
坚持每日一题，加油！

发布于: 1 小时前阅读数: 2

原文链接:【http://xie.infoq.cn/article/310065d0caf418ed9aec8480b】。文章转载请联系作者。

HQ数字卡

关注

还未添加个人签名 2019.09.29 加入

LeetCode,略懂后端的RD

发布

暂无评论

创作场景

【LeetCode】三数之和 Java 题解

题目描述

思路分析

通过代码

总结

HQ数字卡

评论