1. 程式人生 > 其它 >LeetCode No16. 最接近的三數之和

LeetCode No16. 最接近的三數之和

題目

給你一個長度為 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
-10^4 <= target <= 10^4

思路

暴力

暴力大法萬歲,三個for迴圈可以過。

雙指標

這道題目和昨天的三數之和其實是一個思路,找其中一個數,然後用兩個指標分別代表另外兩數,判斷條件由等於改成了相減後絕對值最小。

AC程式碼

暴力

點選檢視程式碼
class Solution {
    
    public int threeSumClosest(int[] nums, int target) {
        int len = nums.length;
        Arrays.sort(nums);
        int res = 100000;
        // 如何剪枝
        for(int first=0; first<len; first++) {
            for(int second = first+1; second<len; second++) {
                // 這層應該是不需要的,有超時風險
                for(int thread = second+1; thread<len; thread++) {
                    int num = nums[first]+nums[second]+nums[thread];
                    if( Math.abs(target-num)<=Math.abs(target-res) ) {
                        res = num;
                    }
                }
            }
        }
        return res;
    }
}

雙指標

點選檢視程式碼
class Solution {
    
    public int threeSumClosest(int[] nums, int target) {
        int len = nums.length;
        Arrays.sort(nums);
        int res = 100000;
        for(int i=0; i<len-2; i++) {
            if( i!=0 && nums[i]==nums[i-1] ) {
                continue;
            }
            int j = i + 1, k = len - 1;
            while (j < k) {
                int sum = nums[i] + nums[j] + nums[k];
                if (sum == target) {
                    return target;
                }
                if (Math.abs(sum - target) < Math.abs(res - target)) {
                    res = sum;
                }
                if (sum > target) {
                    int k0 = k - 1;
                    while (j < k0 && nums[k0] == nums[k]) {
                        --k0;
                    }
                    k = k0;
                } else {
                    // 如果和小於 target,移動 b 對應的指標
                    int j0 = j + 1;
                    // 移動到下一個不相等的元素
                    while (j0 < k && nums[j0] == nums[j]) {
                        ++j0;
                    }
                    j = j0;
                }
            }
        }
        return res;
    }
}