leetcode

最新推荐文章于 2024-07-25 22:25:49 发布

不务正业的野猴子

最新推荐文章于 2024-07-25 22:25:49 发布

阅读量165

点赞数

分类专栏：算法文章标签： leetcode 算法 java

本文链接：https://blog.csdn.net/quan20111992/article/details/131859562

版权

算法专栏收录该内容

3 篇文章 0 订阅

订阅专栏

给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请你返回所有和为 0 且不重复的三元组。
注意：答案中不可以包含重复的三元组。

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
      
        if (nums == null || nums.length < 3) {
            return Collections.emptyList();
        }

        Arrays.sort(nums);

        List<List<Integer>> results = new ArrayList<>();

        for (int i = 0; i < nums.length - 2; i++) {


            if (nums[i] + nums[i+1] + nums[i+2] > 0) {
                break;
            }

            if (nums[i] + nums[nums.length -1] + nums[nums.length-1] < 0) {
                continue;
            }

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

            int l = i + 1;
            int r = nums.length - 1;

            while (l < r) {
                int sum = nums[i] + nums[l] + nums[r];
                if (sum == 0) {
                    
                    results.add(Arrays.asList(nums[i], nums[l], nums[r]));

                    while (l < r && nums[l] == nums[l+1]) {
                        l++;
                    }
                    while (l < r && nums[r] == nums[r-1]) {
                        r--;
                    }
                    l++;
                    r--;
                } else if (sum < 0) {
                    while (l < r && nums[l] == nums[l+1]) {
                        l++;
                    }
                    l++;
                } else if (sum > 0) {
                    while (l < r && nums[r] == nums[r-1]) {
                        r--;
                    }
                    r--;
                } else {
                    continue;
                }
            }
        }

        return results;
    }
}