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;
} 

猜你喜欢

转载自blog.csdn.net/weixin_43824564/article/details/106221982