1. 程式人生 > 實用技巧 >最接近的三數之和

最接近的三數之和

1.問題描述

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

示例:

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

提示:

  • 3 <= nums.length <= 10^3
  • -10^3<= nums[i]<= 10^3
  • -10^4<= target<= 10^4

2.求解

雙指標法

本題與題目

非常類似,這裡我們用相同的思路來解決

  1. 先對陣列進行排序
  2. 遍歷陣列,固定最左邊數作為一個指標k,定義指標ij分別從k和陣列最右端,然後從兩端分別移動ij
  3. 令三數之和等於sum,定義整數ansmax,每次更新target - sum的值,若小於ans則令ans = target - sum
    • sum > 0時,進行j--的操作
    • sum < 0時,進行i++的操作
    • sum = 0時,跳出
程式碼如下
    public int threeSumClosest(int[] nums, int target) {
        int len = nums.length,ans = nums[0] + nums[1] + nums[2] ;
        if(nums == null || len < 3)
            return ans;
        Arrays.sort(nums);
        for (int k = 0; k < len - 2; k++) {
            int i = k + 1, j = len - 1;
            while (i < j) {
                int sum = nums[k] + nums[i] + nums[j];
                if (Math.abs(sum - target) < Math.abs(target - ans)) {
                    ans = sum;
                }
                if(sum > target){
                    j--;
                }
                else if(sum < target) {
                    i++;
                }
                else {
                    return ans;
                }
            }
        }
        return ans;
    }
  • 時間複雜度O(n²)
  • 空間複雜度O(1)