三数之和(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;
    }
}

猜你喜欢

转载自blog.csdn.net/m0_60264772/article/details/121462341