力扣第十六题-最接近的三数之和

250 阅读2分钟

这是我参与更文挑战的第13天,活动详情查看: 更文挑战

前言

力扣第十六题 最接近的三数之和 如下所示:

给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。

示例

输入:nums = [-1,2,1,-4], target = 1
输出:2
解释:与 target 最接近的和是 2 (-1 + 2 + 1 = 2) 。

一、思路

这一题与前面的 第十五题-三数之和 思路上是一样的,还是采用 排序+双指针 来实现的。

思路大致分为以下两步

  • 将nums升序排列
  • 选择前两个元素从左边开始,第三个元素从右边开始,保存最接近的值

在这里以示例为例,描述一下具体的步骤:

假设 nums = [-1,2,1,-4], target = 1,ret为返回结果

  1. 将nums升序排列,即 nums = [-4, -1, 1, 2]
  2. -4 和 -1 作为前两个元素,第三个元素从右向左依次选择。经过此次循环得到结果 (-4) + (-1) + 2 = (-3),即 ret = -3
  3. -4 和 1 作为前两个元素,第三个元素只有2了。此次循环结束后,结果更新为 (-4) + 1 + 2 = (-3),即 ret = -1
  4. -1 和 1 作为前两个元素,第三个元素只有2了。此次循环结束后,结果更新为 (-1) + 1 + 2 = 2,即ret = 2
  5. 返回最靠近 target=1 的结果 2

二、实现

实现代码

为了能够方便的找到最接近的值,所以一开始将结果设为nums数组的前三个元素。

    public int threeSumClosest(int[] nums, int target) {
        int ret = nums[0] + nums[1] + nums[2];
        // 先将数组变为升序的
        Arrays.sort(nums);
        // 采用双指针遍历
        for (int i=0; i<nums.length-2; i++) {
            if (i != 0 && nums[i] == nums[i-1])
                continue;
            int m = nums.length-1;
            for (int j=i+1; j<nums.length-1; j++) {
                if (j != i+1 && nums[j] == nums[j-1])
                    continue;
                // 向左移动指针,且右指针在左指针的右边
                while(j < m && nums[i]+nums[j]+nums[m]>target) {
                    // 取最接近的值
                    if (Math.abs(ret - target) >= Math.abs(nums[i]+nums[j]+nums[m] - target)) {
                        ret = nums[i]+nums[j]+nums[m];
                    }
                    m = m-1;
                }
                if (j == m) {
                    break;
                }
                // 取最接近的值
                if (Math.abs(ret - target) >= Math.abs(nums[i]+nums[j]+nums[m] - target)) {
                    ret = nums[i]+nums[j]+nums[m];
                }
                if (ret == target) {
                    return ret;
                }
            }
        }
        return ret;
    }

测试

    public static void main(String[] args) {
        int ret = new Number16().threeSumClosest(new int[]{1,2,5,10,11}, 12);
        System.out.println(ret);
    }

结果

image.png

三、总结

感谢看到最后,非常荣幸能够帮助到你~♥