大佬说是三分裸题 于是我补了一下 确实挺裸的
关于高度 显然是有一个cost 的峰值函数的 于是我们三分一个高度 然后进行计算
记住计算的时候 要注意一个贪心 就是 当M<A+R的时候 我们先进行M操作
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e5+10;
int n;
ll A,R,M;
ll H[N];
ll checkmid(ll h){
ll add=0,sub=0;
for(int i = 1; i <= n; i++){
if(H[i]<h) add+=h-H[i];
else sub+=H[i]-h;
}
if(M<A+R){
ll m=min(add,sub),ret=0;
ret+=m*M;
if(add>sub) ret+=(add-sub)*A;
else ret+=(sub-add)*R;
return ret;
}else return add*A+sub*R;
}
int main(){
scanf("%d%lld%lld%lld",&n,&A,&R,&M);
ll l = 0,r = 0;
for(int i = 1; i <= n; i++) scanf("%lld",&H[i]),r=max(r,H[i]);
while(l<r){
ll lans=l+(r-l)/3,rans=r-(r-l)/3;
if(checkmid(lans)>=checkmid(rans)) l=lans+1;
else r=rans-1;
}
printf("%lld\n",checkmid(l));
return 0;
}