算法实例分析:使数组相等的最小开销

使数组相等的最小开销

通过题意分析可知要让所有值相等,必然不需要超出数据的最大最小值,因此左右边界可以预先缩小范围。然后根据我们上面的分析不断缩小搜索边界范围。

关于函数的计算,只要统计所有数据与的差值再乘上权重即可。

最后注意,为了保证答案的正确性需要再在两端取一次最值

复制代码
class Solution {
public:
    long long minCost(vector<int>& nums, vector<int>& cost) {
        auto sum = [&](int k) -> long long {
            long long ans = 0;
            for (int i = 0; i < nums.size(); i += 1) {
                ans += 1LL * cost[i] * abs(nums[i] - k);
            }
            return ans;
        };

        int left  = *min_element(nums.begin(), nums.end());
        int right = *max_element(nums.begin(), nums.end());
        while (left <= right) {
            int lmid = left + (right - left) / 3;
            int rmid = right - (right - left) / 3;
            /// 凹区间
            if (sum(lmid) <= sum(rmid)) {
                right = rmid - 1;
            } else {
                left = lmid + 1;
            }
        }

        // 最终取两点间的最值
        return min(sum(left), sum(right));
    }
};

总结

再回头看一下二分与三分的书写形式会发现两者是如此的相像。因此我们要着重注意两者的区别:

本文只是列举了两个典型的简单例子,各种应用场景对算法的具体细节有所差异。

但是万变不离其宗,只要理清搜索的实际的搜索流程就不需要僵硬地套用所谓的模板。

相关推荐
北域码匠几秒前
高通滤波算法深度解析(High-Pass Filter)
stm32·算法·c#·数字信号处理·嵌入式开发·滤波算法·高通滤波
指掀涛澜天下惊30 分钟前
强化学习进阶篇八 策略梯度算法
深度学习·学习·算法·强化学习
乌萨奇也要立志学C++35 分钟前
【洛谷】kmp算法
开发语言·算法
禹凕43 分钟前
Dijkstra算法详解与应用
python·算法
YaraMemo2 小时前
元启发式算法框架
人工智能·算法·5g·信息与通信·启发式算法·信号处理
weixin_307779132 小时前
一维无粘 Burgers 方程的激波形成问题:MacCormack 格式求解
c++·算法·matlab
HugoStudio_SWAN2 小时前
洛谷 B4500 / B4449 / B3843 凯撒密码、密码强度与密码合规——加密与安全的三道门
c++·学习·程序人生·算法·安全
枫叶林FYL3 小时前
【群体智能集群控制工程实践】第3章 一致性协同算法
算法
禹凕4 小时前
滑动窗口算法实战指南
python·算法
DisonTangor4 小时前
【腾讯混元雪耻归来】 Hy4 preview:770B 参数 MoE 旗舰模型,1M 上下文全面开源
人工智能·算法·开源·aigc·腾讯云·腾讯云ai代码助手