0
点赞
收藏
分享

微信扫一扫

力扣最热100题之三数之和

霸姨 2022-03-30 阅读 65

题目:输入一个数组,找出数组中所有的三个数和为0的组合
分析:
1.先排序,对数组进行排序。
2.然后使用双指针法。首先固定一个元素,循环遍历固定元素,然后从固定元素的下一个开始定义 左右指针,如果三者加和大于0,则右指针向左移,小于0,则左指针向右移,如果固定的元素直接就大于0了,因为刚开始数组排好序的,如果此时固定元素就大于0了,那后面元素就都大于0,所以后面的三数组合之和不可能在小于0,那就说明没必要再次寻找了。

需要注意的点:1.记得要先排序,排好序的数组更方便用于去重,因为重复元素都排列在一起了。2.要注意左右指针的去重操作,不论是大于0,还是小于0,甚至是等于0,在移动指针时都要注意去重,可以大大加快程序的运行速度。

public List<List<Integer>> threeSum(int[] nums) {

        Arrays.sort(nums);

        List<List<Integer>> res = new ArrayList<>();
        // 第一个元素为k,固定,然后从第二个元素开始,到length-1个元素结束,一共length-2个元素
        for(int k = 0; k < nums.length - 2; k++){

            // 因为该数组排好序了,所以此时如果k都大于0,那么后面的元素肯定大于0,证明搜索结束
            if(nums[k] > 0)
                break;

            // 证明数组中有数据重复,跳过重复数据
            if(k > 0 && nums[k] == nums[k - 1])
                continue;

            // 从固定元素的下一个开始查找
            int i = k + 1, j = nums.length - 1;

            while(i < j){

                int sum = nums[k] + nums[i] + nums[j];

                if(sum < 0){

                    while(i < j && nums[i] == nums[++i]);

                } else if (sum > 0) {

                    while(i < j && nums[j] == nums[--j]);

                } else {

                    res.add(new ArrayList<Integer>(Arrays.asList(nums[k], nums[i], nums[j])));

                    while(i < j && nums[i] == nums[++i]);

                    while(i < j && nums[j] == nums[--j]);

                }
            }
        }
        return res;
    }
举报

相关推荐

0 条评论