Leetcode1300

求整数k,使得原数组比k大的值都变为k,整个数组的和与目标值target的绝对值差最小。

 

注意到:k最大只能是max(a[0], a[1]...), 再大的加话,原数组的值也不会变化,最小的是1

我们可以二分来解决这个问题

我们维护的区间(l, r)求出使得数组和比target大的最小k,

但是最优解可能是比target小一点或大一点

就是说

public class Leetcode1300 {

    public int findBestValue(int[] arr, int target) {
        int l = 1;
        int r = 0;
        int s = 0;

        for (int x : arr) {
            s += x;
            r = Math.max(r, x);
        }
        if (s == target) return r;
        while (l < r) {
            int mid = (l + r) / 2;
            int sum = getSum(arr, mid);
            if (sum == mid) return mid;
            if (sum > target) r = mid;
            else l = mid + 1;
        }
        if (Math.abs(getSum(arr, l - 1) - target) <= Math.abs(getSum(arr, l) - target)) {
            return l - 1;
        } else {
            return l;
        }
    }

    int getSum(int[] arr, int mid) {
        int s = 0;
        for (int x : arr) {
            s += x > mid ? mid : x;
        }
        return s;
    }
}

可能k-1的时候才是最优解,最后check下就好了

你可能感兴趣的:(二分,leetcode)