LeeCode 1300 三分 / 暴力

最新推荐文章于 2024-04-29 23:15:35 发布

SHOHOKUKU

最新推荐文章于 2024-04-29 23:15:35 发布

阅读量100

点赞数

分类专栏：基本算法

本文链接：https://blog.csdn.net/neweryyy/article/details/106753776

版权

基本算法专栏收录该内容

252 篇文章 1 订阅

订阅专栏

题意

传送门 LeeCode 1300. 转变数组后最接近目标值的数组和

题解

三分

随着 $v a l u e$ 减小，数组和非严格递减。那么 $t a r g e t$ 与数组和的绝对值在 $v a l u e$ 附近会出现极小值，三分法求解即可。

class Solution
{
public:
    vector<int> arr;
    int target;
    int calc(int v)
    {
        int sum = 0;
        for (int x : arr)
            sum += x > v ? v : x;
        return abs(sum - target);
    }
    int findBestValue(vector<int> &arr, int target)
    {
        this->arr = arr, this->target = target;
        int lb = -1, ub = 1e5 + 1;
        while (ub - lb > 1)
        {
            int mid = (ub + lb) >> 1, mmid = (mid + ub) >> 1;
            if (calc(mid) <= calc(mmid))
                ub = mmid;
            else
                lb = mid;
        }
        if (calc(lb) == calc(ub))
            return lb;
        return calc(lb) > calc(ub) ? ub : lb;
    }
};

暴力

对 $a r r$ 排序后，设实数 $v a l u e^{'}$ 满足下式

$abs(target-\{\sum\limits_{i=0}^{k-1}arr[k]+value'\times (n-k)\})=0$

此时

$value'=(target-\sum\limits_{i=0}^{k-1}arr[k])/(n-k)$

遍历一遍 $a r r$ ，找到满足 $arr[k-1]\leq value'\leq arr[k]$ 的 $k$ ，计算出 $v a l u e^{'}$ 后四舍五入即为 $v a l u e$ 。

class Solution
{
public:
    int findBestValue(vector<int> &arr, int target)
    {
        sort(arr.begin(), arr.end());
        int sum = 0, n = arr.size();
        for (int i = 0; i < n; i++)
        {
            int v = (target - sum) / (n - i);
            if (v < arr[i])
            {
                double vv = (target - sum) / (double)(n - i);
                if (vv - v > 0.5)
                    return v + 1;
                return v;
            }
            sum += arr[i];
        }
        return arr[n - 1];
    }
};