LeetCode16 最接近的三数之和

• 题目描述

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

返回这三个数的和。

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

链接：https://leetcode.cn/problems/3sum-closest

• 示例

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

输出: 2

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

 

• 算法思想

首先对输入的数据进行排序,然后固定一个数作为第一个要相加的数,并在其余的数的区间采用双指针法,得到三个数之后,判断三数之和是否比上次更邻近target,若更接近更新相应的数据.

 

• 代码

#include<iostream>
#include<algorithm>
using namespace std;
int main() {
    int n;
    cin >> n;
    int* a = new int[n];
    for (int i = 0; i < n; i++)
        cin >> a[i];
    int target;
    cin >> target;
    sort(a, a + n);//默认升序排序
    int dis = INT_MAX;//初始值为最大值
    int sum = 0;
    int res = 0;
    for (int i = 0; i <= n - 3; i++) {
        //双指针法
        int l = i + 1, r = n - 1;
        while (l < r) {
            sum = a[i] + a[l] + a[r];
            int x = abs(sum - target);
            if (x< dis) {//间距比上次更小，更新dis，res
                dis = x;
                res = sum;
            }
            if (sum > target) r--;  //三数之和大于目标值，r--,因为右方的数较大
            else if (sum < target) l++; //三数之和小于目标值，l++,因为左方的数较小
            else {   //三数之和等于目标值，直接输出目标值
                cout <<target << endl;
                return 0;
            }
        }
    }
    cout << res << endl;//输出最接近target的数
    return 0;
}