Codeforces Round #643 (Div. 2) 1355E. Restorer Distance 三分

最新推荐文章于 2026-01-02 10:17:02 发布

原创最新推荐文章于 2026-01-02 10:17:02 发布 · 232 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法

codeforces 同时被 2 个专栏收录

19 篇文章

订阅专栏

思维

11 篇文章

订阅专栏

这篇博客探讨了一种解决n堆砖块高度调整问题的算法，通过三种操作（增加、减少和转移砖块）使得所有堆高度相同，目标是最小化操作成本。文章中给出了算法实现，包括检查不同高度的函数，并使用三分搜索找到最优解。最终，根据给定的成本参数a、r和m，计算出最小化总费用。

原题链接
题意：
n堆由相同砖块堆成的堆，有三种操作：
1.在第i堆放一块转，花费a。
2.在第i堆拿走一块转，花费r。
3.把第i堆的砖拿到第j堆，花费m。
求使得所有堆高度相同的最小化费。

思路：
先增后减的函数曲线，三分高度算花费。

#include<bits/stdc++.h>
#define LL long long
#define INF INT64_MAX
#define MOD 998244353
#define ls rt << 1
#define rs rt << 1 | 1
using namespace std;
typedef pair<int,int>pa;
const int N = 1e5+7;
LL h[N], n, a, r, m;
char s[N];
LL check(LL x){
    LL sum1 = 0, sum2 = 0, res = 0;
    for(int i = 1;i <= n;i++){
        if(h[i] > x) sum1 += h[i]-x;
        else sum2 += x-h[i];
    }
    if(sum1>sum2){
        res += sum2*m;
        res += (sum1-sum2)*r;
    }
    else{
        res += sum1*m;
        res += (sum2-sum1)*a;
    }
    return res;
}
int main(){
    scanf("%lld%lld%lld%lld", &n, &a, &r, &m);
    for(int i = 1;i <= n;i++){
        scanf("%lld", &h[i]);
    }
    if(a+r<m) m = a+r;
    LL l = 0, r = 1e9;
    while(l<r){
        LL lmid = l + (r-l)/3;
        LL rmid = r - (r-l)/3;
        if(check(lmid) > check(rmid)) l = lmid+1;
        else r = rmid-1;
    }
    printf("%lld\n", check(l));
    return 0;
}