Java力扣题解:15. 三数之和

题目

分析

代码


题目

给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。

注意:答案中不可以包含重复的三元组。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

分析

顺着题目的意思来即可。首先做一个排序,让数组有序。从头开始遍历,i 位置作为第一个值,同时保证后续枚举时跳过重复的值。内部嵌套一个循环是从 i+1 位置开始,同样后续枚举跳过重复值。确定了两个数之后,可以知道第三个我们要找的值。然后从后往前寻找这个值。记录三个值即可。

一个小提示:下方代码耗时极长,你知道问题出在哪吗?

代码

class Solution {
    public List> threeSum(int[] nums) {
        int length = nums.length;
        List> res = new ArrayList>();
        if(length < 3){
            return res;
        }
        Arrays.sort(nums);
        for(int i = 0; i < length-2; i++){
            int a = nums[i];
            if(i > 0 && a == nums[i-1]){
                continue;
            }
            for(int j = i+1; j < length-1; j++){
                int b = nums[j];
                if(b == nums[j-1] && j-1 != i){
                    continue;
                }
                int cpl = length-1;
                int obj = -a-b;
                while(cpl != j && nums[cpl] >= obj){
                    if(nums[cpl] == obj){
                        List list = new ArrayList<>();
                        list.add(a);
                        list.add(b);
                        list.add(nums[cpl]);
                        res.add(list);
                        break;
                    }
                    else{
                        cpl--;
                    }
                }
            }
        }
        return res;
    }
}

你可能感兴趣的:(leetcode,leetcode)