代码源：CF 1355E（整数三分）

传送

点击查看代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int a[100010];
int n,A,R,M;
ll f(int x)
{
	ll y=0,z=0;//y表示需要增加的次数,z表示需要减少的次数
	for(int i=0;i<n;++i)
	{
		if(a[i]<x) y+=x-a[i];
		else z+=a[i]-x;
	}
	ll t=min(y,z);
	ll res=t*M+(y-t)*A+(z-t)*R;
	return res;
}

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cin>>n>>A>>R>>M;
	M=min(M,A+R);
	for(int i=0;i<n;++i) cin>>a[i];
	int L=0,R=1e9;//三分不需要改变L,R
	while(L+2<R)
	{
		int M1=L+(R-L)/3;
		int M2=L+(R-L)/3*2;

		if(f(M1)>f(M2)) L=M1;//如果M1情况更遭缩小L
		else R=M2;
	}
	ll ans=f(L);
	for(int i=L+1;i<=R;++i) ans=min(ans,f(i));
	cout<<ans<<'\n';
}