(LeetCode 16) 最接近的三數之和

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

示例:
輸入:nums = [-1,2,1,-4], target = 1
輸出:2
解釋:與 target 最接近的和是 2 (-1 + 2 + 1 = 2) 。

解題思路:
排序和雙指針

  1. 數組排序
  2. 數組遍歷,形成一個固定值nums[i]
  3. 設置前指針start = i+1,後指針 end = nums.length -1
  4. 根據 sum = nums[i] + nums[start] + nums[end] 的結果,判斷 sum 與目標 target 的距離,如果更近則更新結果 ans
  5. 同時判斷 sum 與 target 的大小關係,因爲數組有序,如果 sum > target 則 end--,如果 sum < target 則 start++,如果 sum == target 則說明距離爲 0 直接返回結果
    public static 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;
            int end = nums.length - 1;
            while (start < end) {
                int sum = nums[i] + nums[start] + nums[end];
                if (Math.abs(sum - target) < Math.abs(ans - target)) {
                    ans = sum;
                }
                if (sum > target) {
                    end--;
                } else if (sum < target) {
                    start++;
                } else {
                    return ans;
                }

            }
        }
        return ans;
    }
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章