CF Round #643 (Div. 2) E - Restorer Distance

大佬說是三分裸題 於是我補了一下 確實挺裸的

關於高度 顯然是有一個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;
}