LeetCode - 最接近的三數之和

最接近的三數之和
你一個長度爲 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

import java.util.Arrays;

public class Main {
    public static void main(String[] args) {
        int[] nums = {-1, 2, 1, -4};
        if (nums.length < 3) {
            System.out.println("不符合題意的數組大小");
            return;
        }
        int threeSumClosest = threeSumClosest(nums, 1);
        System.out.println(threeSumClosest);
    }

    /**
     * 求出最接近目標值target的三數之和
     *
     * @param nums   要傳入的數組
     * @param target 目標值
     * @return 最接近目標值target的三數之和
     */
    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++) {
            // 前置位置startIndex,後置位置endIndex
            int startIndex = i + 1, endIndex = nums.length - 1;
            while (startIndex < endIndex) {
                int sum = nums[startIndex] + nums[endIndex] + nums[i];
                // 更新最接近的值,通過比較,獲取到最接近target的值
                if (Math.abs(target - sum) < Math.abs(target - ans)) {
                    ans = sum;
                }
                // 如果求和值大於目標值,後置位置往前挪
                if (sum > target) {
                    endIndex--;
                } else if (sum < target) {
                    // 如果求和值小於目標值,後置位置往前挪
                    startIndex++;
                } else {
                    // 如果等於目標值,直接返回
                    return ans;
                }
            }
        }
        return ans;
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章