畫解算法：16. 最接近的三數之和

題目連接

leetcode-cn.com/problems/3s…

題目描述

給定一個包括 n 個整數的數組 nums 和 一個目標值 target。找出 nums 中的三個整數，使得它們的和與 target 最接近。返回這三個數的和。假定每組輸入只存在惟一答案。

例如，給定數組 nums = [-1，2，1，-4], 和 target = 1.

與 target 最接近的三個數的和爲 2. (-1 + 2 + 1 = 2).
複製代碼

解題方案

思路

• 標籤：排序和雙指針
• 本題目由於要計算三個數，若是靠暴力枚舉的話時間複雜度會到O(n^3)，須要下降時間複雜度
• 首先進行數組排序，時間複雜度O(nlogn)
• 在數組nums中，進行遍歷，每遍歷一個值利用其下標i，造成一個固定值nums[i]
• 再使用前指針指向start = i + 1處，後指針指向end = nums.length - 1處，也就是結尾處
• 根據 sum = nums[i] + nums[start] + nums[end] 的結果，判斷sum與目標target的距離，若是更近則更新結果ans
• 同時判斷sum與target的大小關係，由於數組有序，若是sum > target 則 end--，若是sum < target 則 start++，若是sum == target 則說明距離爲0直接返回結果
• 整個遍歷過程，固定值爲n次，雙指針爲n次，時間複雜度爲O(n^2)
• 總時間複雜度：O(nlogn) + O(n^2) = O(n^2)

代碼

class Solution {
    public int threeSumClosest(int[] nums, int target) {
        Arrays.sort(nums);
        int ans = nums[0] + nums[1] + nums[2];
        for(int i=0;i<nums.length;i++) {
            int start = i+1, end = nums.length - 1;
            while(start < end) {
                int sum = nums[start] + nums[end] + nums[i];
                if(Math.abs(target - sum) < Math.abs(target - ans))
                    ans = sum;
                if(sum > target)
                    end--;
                else if(sum < target)
                    start++;
                else
                    return ans;
            }
        }
        return ans;
    }
}
複製代碼

畫解

後臺回覆「算法」，加入每天算法羣 以爲算法直擊靈魂，歡迎點擊在看和轉發