算法题:三数之和

一、题目

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

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

二、思路

先对数组排序,然后利用双指针,利用排序数组的两数之和的思想,即可解决。

三、实现

public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> result = new ArrayList<>();

        Arrays.sort(nums);

        for (int i = 0; i < nums.length; i++) {
            if (nums[i] > 0) {
                break;
            }
            if (i > 0 && nums[i] == nums[i - 1]) {
                continue;
            }

            int j = i + 1;
            int k = nums.length - 1;

            int target = 0 - nums[i];

            while (j < k) {
                if (target == nums[j] + nums[k]) {
                    List<Integer> list = new ArrayList<>();
                    list.add(nums[i]);
                    list.add(nums[j]);
                    list.add(nums[k]);
                    result.add(list);

                    while (j < k && nums[j] == nums[j + 1]) {
                        j++;
                    }
                    while (j < k && nums[k] == nums[k - 1]) {
                        k--;
                    }

                    j++;
                    k--;
                } else if (target > nums[j] + nums[k]) {
                    j++;
                } else {
                    k--;
                }
            }

        }

        return result;
    }

发布了83 篇原创文章 · 获赞 0 · 访问量 4530

猜你喜欢

转载自blog.csdn.net/zhangdx001/article/details/105406073