[数据结构与算法]三数之和（leetcode每日打卡）

目录

题目描述

思路

题解

题目描述

给你一个包含 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]
输出：[]

思路

来自

排序 + 双指针
本题的难点在于如何去除重复解。

算法流程：
特判，对于数组长度 n，如果数组为null 或者数组长度小于 3，返回 []。
对数组进行排序。
遍历排序后数组：
若 nums[i]>0：因为已经排序好，所以后面不可能有三个数加和等于 0，直接返回结果。
对于重复元素：跳过，避免出现重复解
令左指针 L=i+1，右指针 R=n?1，当L<R 时，执行循环：
当 nums[i]+nums[L]+nums[R]==0，执行循环，判断左界和右界是否和下一位置重复，去除重复解。并同时将 L,R 移到下一位置，寻找新的解
若和大于 0，说明nums[R] 太大，R 左移
若和小于 0，说明 nums[L] 太小，L 右移
复杂度分析
时间复杂度：O(n^2)
空间复杂度：O(1)

题解

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> lists=new ArrayList<List<Integer>>();
        Arrays.sort(nums);

        for(int i=0;i<nums.length;i++){
            if(nums[i]>0){return lists;}
            if(i>0&&nums[i]==nums[i-1])continue;
            int cur=nums[i];
            int L=i+1;
            int R=nums.length-1;
            while(L<R){ 
                   int sum=cur+nums[L]+nums[R];    
                     
                if(sum==0){  List<Integer> list=new ArrayList<>();
                    list.add(cur);
                    list.add(nums[L]);
                    list.add(nums[R]);
                    lists.add(list);
                    while(L<R&&nums[L+1]==nums[L]){++L;}
                    while(L<R&&nums[R-1]==nums[R]){--R;}
                    ++L;
                    --R;
                }else if(sum<0){
                    ++L;
                }else{
                    --R;
                }
            }
        }
        return lists;
    }
}