【探索-中级算法】三数之和

【探索-中级算法】三数之和_第1张图片

参考链接:LeetCode总结-K-Sum问题


本文介绍的解题思想的核心就是排序,排序有两个目的,第一个是次要的,即方便排除重复的组合。第二个就是使得可以按照递增或者递减方便的移动指针 l、r

在排序之后,就可以对数组进行遍历,目标就是找到符合 nums[l] + nums[r] == -target 的组合,而 target 是固定不变的,只要动态的寻找符合条件的 nums[l] 、nums[r]

public List<List<Integer>> threeSum(int[] nums) {
    Arrays.sort(nums);
    List<List<Integer>> result = new ArrayList<>();
    for (int i = 0; i < nums.length - 2; i++) {
        if (i > 0 && nums[i] == nums[i - 1]) { //跳过相同的数以避免相同组合的产生
            //比如在 -1 -1 0 1 2 的 nums 数组中
            // target 为 nums[0] 与 nums[1] 的效果肯定是一样的
            continue;
        }
        find(result, nums, i + 1, nums.length - 1, nums[i]);
    }
    return result;
}

public void find(List<List<Integer>> result, int[] nums, int start, int end, int target) {
    int l = start, r = end;
    while (l < r) {
        if (target + nums[l] + nums[r] == 0) {
            List<Integer> list = new ArrayList<>();
            list.add(target);
            list.add(nums[l]);
            list.add(nums[r]);
            result.add(list);
            while (l < r && nums[l] == nums[l + 1]) ++l;//跳过相同的数以避免相同组合的产生
            while (l < r && nums[r] == nums[r - 1]) --r;//跳过相同的数以避免相同组合的产生
            ++l;
            --r;
        } else if (target + nums[l] + nums[r] < 0) {
        	// nums[l]+nums[r] < -target,因此还需加大 nums[l]+nums[r] 的和
            ++l;
        } else { 
        	// 同上
            --r;
        }
    }
}

需要注意的是,题目中说要避免重复的三元组。

你可能感兴趣的:(LeetCode,渣渣的算法历练之路)