刷题:最接近的三数之和

154 阅读1分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第30天,点击查看活动详情

题目来源: 16. 最接近的三数之和

题目:

给你一个长度为 n 的整数数组 nums 和 一个目标值 target。请你从 nums 中选出三个整数,使它们的和与 target 最接近。

返回这三个数的和。

假定每组输入只存在恰好一个解。

示例 1:

输入:nums = [-1,2,1,-4], target = 1

输出:2

解释:与 target 最接近的和是 2 (-1 + 2 + 1 = 2) 。

示例 2:

输入:nums = [0,0,0], target = 1

输出:0  

提示:

  • 3 <= nums.length <= 1000

  • 1000 <= nums[i] <= 1000

  • 104 <= target <= 104

思路分析:

  • 三数之和问题,依旧使用排序和双指针
  • 在数组 nums 中,进行遍历,每遍历一个值利用其下标i,形成一个固定值 nums[i]
  • 初始化一个用于保存结果的值 result = nusm[0] + nums[1] + nums[2] (不要自己设初值,直接从数组中抽取三个元素,假设这是最接近的三数之和,然后再更新就是了)。
  • 遍历数组,遍历是下标为i。双指针:前指针指向 start = i + 1 处,后指针指向 end = nums.length - 1 处,也就是结尾处。
  • 假设结果为前三个数之和,即result = nums[0] + nums[1] + nums[2]。设计根sum = nums[i] + nums[start] + nums[end] ,通过判断 sum 与目标 target的距离,如果更近,就更新结果result。
  • 移动双指针
  • 判断 sum 与 target 的大小关系,因为数组有序,如果 sum > target 则 end–,如果 sum < target 则 start++,如果 sum == target 则说明距离为 0 直接返回结果。
  • 重复上面的操作,直到i循环结束为止,返回 res

AC代码

class Solution {
public:
    int threeSumClosest(vector<int>& nums, int target) {
        sort(nums.begin(),nums.end());
        int sum = accumulate(nums.begin(),nums.begin()+3,0);
        for(int i = 0; i < nums.size() - 2; ++i){
            if(i > 0 && nums[i] == nums[i-1]) continue;
            int end = nums.size()-1,j = i + 1;
            while(j < end){
                int res = nums[i] + nums[j] + nums[end];
                if(abs(res - target) < abs(sum - target)){
                    sum = res;
                }
                if(res > target){
                    --end;
                    while(end != j && nums[end] == nums[end+1]) --end;
                }
                else if(res < target){
                    ++j;
                    while(j != end && nums[j] == nums[j-1])  ++j;
                }
                else return res;
            }
        }
        return sum;
    }
};

image.png