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

使数组相等的最小开销

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

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

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

复制代码
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));
    }
};

总结

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

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

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

相关推荐
2501_906565122 小时前
数学之美探究
算法
oier_Asad.Chen2 小时前
【洛谷题解/AcWing题解/OI学习笔记】洛谷P2868【USACO07DEC】Sightseeing Cows G(01分数规划求最大比率)
算法·图论·spfa·二分·负环
leisoo80974 小时前
财报数据怎么排雷本地化Python构建财务异常预警系统
人工智能·python·算法
luj_17686 小时前
塔防牌:策略与卡牌的智慧碰撞
服务器·c语言·开发语言·经验分享·算法
郝学胜-神的一滴7 小时前
干货版《算法导论》17:二叉树核心原理、遍历逻辑与高阶实操全解
数据结构·c++·python·算法·计算机·编程
爱编程的小新☆7 小时前
【LeetCode】从递归到 Flood Fill:5 道题吃透 DFS 的选择、回溯与标记
java·算法·leetcode·深度优先·回溯·flood fill
Water_Sunzhipeng7 小时前
2024牛客暑期多校训练营1
算法
hetao17338377 小时前
2026-08-09~08-14 hetao1733837 的刷题记录
c++·算法
技术小黑7 小时前
RNN算法实战系列06 | LSTM 实现糖尿病探索与预测
rnn·算法·lstm
alphaTao7 小时前
LeetCode 每日一题 2026/8/10-2026/8/16
算法·leetcode