LeetCode 16 3Sum Closest

LeetCode 16 3Sum Closest

思路

本题是3Sum的变种。在内层循环中，根据sum和target的大小，调整j指向的位置和k指向的位置。内层循环每次只调整j或k中的一个。

代码

class Solution {
public:
    int threeSumClosest(vector<int>& nums, int target) {
        sort(nums.begin(), nums.end());
        map<int, int> mp;
        int sum;
        
        for (int i = 0; i < nums.size(); ++i) {
            int j = i + 1, k = nums.size() - 1;
            while (j < k) {
                sum = nums[i] + nums[j] + nums[k];
                mp[abs(sum - target)] = sum;
                if (sum < target) {
                    while (j < k && nums[j+1] == nums[j]) ++j;
                    ++j;
                }
                else if (sum > target) {
                    while (j < k && nums[k-1] == nums[k]) --k;
                    --k;
                }
                else return target;
            }
            
            while (i + 1 < nums.size() && nums[i+1] == nums[i]) ++i;
        }
        
        return mp.begin()->second;
    }
};