LeetCode 16 3Sum Closest（最接近的3個數的和）

LeetCode 16 3Sum Closest（最接近的3個數的和）

翻譯

給定一個有n個整數的數組S，找出S中3個數，使其和等於一個給定的數，target。

返回這3個數的和，你可以假定每個輸入都有且只有一個結果。

例如，給定S = {-1 2 1 -4}，和target = 1。

那麼最接近target的和是2。(-1 + 2 + 1 = 2)。

原文

Given an array S of n integers, 
find three integers in S such that the sum is closest to a given number, target. 

Return the sum of the three integers. 
You may assume that each input would have exactly one solution.

For example, given array S = {-1 2 1 -4}, and target = 1.

The sum that is closest to the target is 2. (-1 + 2 + 1 = 2).

思考

也許我已經開始體會到上一題中別人寫的方法的思想了。

在這個題目中，我們要做以下幾件事：

用sort對輸入的數組進行排序

求出長度len，current之所以要小於len−2，是因為後面需要留兩個位置給front和back

始終保證front小於back

計算索引為current、front、back的數的和，分別有比target更小、更大、相等三種情況

更小：如果距離小於close，那麼close便等於target−sum，而結果就是sum。更大的情況同理

如果相等，那麼要記得將0賦值給close，result就直接等於target了

隨後為了避免計算重復的數字，用三個do/while循環遞增或遞減它們

代碼

class Solution
{
public:
    int threeSumClosest(vector& nums, int target) {
        sort(nums.begin(), nums.end());
        int len = nums.size();
        int result = INT_MAX, close = INT_MAX;
        for (int current = 0; current < len - 2; current++) {
            int front = current + 1, back = len - 1;
            while (front < back) {
                int sum = nums[current] + nums[front] + nums[back];
                if (sum < target) {
                    if (target - sum < close) {
                        close = target - sum;
                        result = sum;
                    }
                    front++;
                }
                else if (sum > target) {
                    if (sum - target < close) {
                        close = sum - target;
                        result = sum;
                    }
                    back--;
                }
                else {
                    close = 0;
                    result = target;
                    do {
                        front++;
                    } while (front < back&&nums[front - 1] == nums[front]);
                    do {
                        back--;
                    } while (front < back&&nums[back + 1] == nums[back]);
                }
            }
            while (current < len - 2 && nums[current + 1] == nums[current]) {
                current++;
            }
        }
        return result;
    }
};

和本道題關聯密切的題目推薦：

傳送門：LeetCode 15 3Sum（3個數的和）
傳送門：LeetCode 18 4Sum（4個數的和）