leetcode解题报告16. 3Sum Closest-优快云博客

leetcode解题报告16. 3Sum Closest

难度是medium

题目描述

和3SUM问题类似，输入一个整数数组，和一个目标整数target。从数组中选择3个数，使得这三个数之和最接近target。要求返回这三个数之和即可，不要求具体返回是哪三个数。默认有且只有一个解。

我的思路

同样地最简单的方法是三层循环暴力遍历即可。但是这一般不是我们相要的答案。
和3sum问题相比，其实我们可以发现这个问题是比3sum问题更简单的。同样是遍历所有三元组，这个问题不需要考虑三元组去重的问题，而且一旦找到等于target的结果即可返回，不要全部遍历完。
所以可以直接套用解决3sum问题时的遍历办法进行遍历。首先为了更好地遍历，对输入数组进行排序，而且我们假定三元组中元素是升序的，具体遍历时，先固定第一个元素(最小元素)，再通过左右指针从数组的头尾分别遍历第二和第三个元素。如果当前结果相对target偏小，那么左指针向右移动，如果偏大，那么右指针向左移动，直到左右指针相遇。在整个遍历过程记录最接近的结果。如果遇到等于target的情况，则可以直接返回。
具体看代码。

我的代码

#include <algorithm>

using namespace std;

class Solution {
public:
    int threeSumClosest(vector<int>& nums, int target) {
        vector<int> my_nums(nums.begin(), nums.end());
        sort(my_nums.begin(), my_nums.end());
        int cur_min = 1000000;
        int ans;
        for (int i = 0; i < my_nums.size() - 2; i++) {
            int left = i + 1;
            int right = my_nums.size() - 1;
            int new_target = target - my_nums[i];
            while (left != right) {
                int cur = new_target - my_nums[left] - my_nums[right];
                if (cur > 0) {
                    left++;
                    if (cur < cur_min) {
                        cur_min = cur;
                        ans = target - cur;
                    }
                } else if (cur < 0) {
                    right--;
                    if (-cur < cur_min) {
                        cur_min = -cur;
                        ans = target - cur;
                    }
                } else {
                    return target;
                }
            }
        }
        return ans;
    }
};