题目传送门
题意 给你一个长度为n的数组,你可以向一个元素加1,代价为A,也可以向一个元素减1,代价为R,还可以把一个元素的1移动到另外一个元素上,代价为M,现在要求你做一些操作,使得所有元素高度相等。
思路
根据题意,合法答案就应该在min(数组元素)~ max(数组元素)之间,因为如果全加或者全减,到这里就停了,再加或者再减费用更高。我们发现答案在这之间,那么到两端的费用可能要大于真正的答案,答案不具有单调性,不能二分高度,但是它好像一个单峰函数,我们三分找最小即可。注意三分判断的时候,贪心的去使得元素相同,就要判断A、R、M的关系,得出最优策略。
#include<bits/stdc++.h>
#define endl '\n'
#define null NULL
#define ls p<<1
#define rs p<<1|1
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define ll long long
#define int long long
#define vi vector<int>
#define mii map<int,int>
#define pii pair<int,int>
#define ull unsigned long long
#define all(x) x.begin(),x.end()
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define ct cerr<<"Time elapsed:"<<1.0*clock()/CLOCKS_PER_SEC<<"s.\n";
char *fs,*ft,buf[1<<20];
#define gc() (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<20,stdin),fs==ft))?0:*fs++;
inline int read(){int x=0,f=1;char ch=gc();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=gc();}
return x*f;}
using namespace std;
const int N=1e6+5;
const int inf=0x7fffffff;
const int mod=998244353;
const double eps=1e-6;
int n,a,r,m,h[N];
int check(int mid)
{
int res=0,sum1=0,sum2=0;
if(a+r<=m)
{
for(int i=1;i<=n;i++)
{
if(h[i]<mid)
res+=(mid-h[i])*a;
else
res+=(h[i]-mid)*r;
}
}
else
{
for(int i=1;i<=n;i++)
{
if(h[i]<mid)
sum1+=(mid-h[i]);
if(h[i]>mid)
sum2+=(h[i]-mid);
}
int t=min(sum1,sum2);
res+=t*m;
sum1-=t;sum2-=t;
res+=sum1*a+sum2*r;
}
return res;
}
signed main()
{
cin>>n>>a>>r>>m;
for(int i=1;i<=n;i++)
cin>>h[i];
int l=0,r=1e9;
while(l+20<r)
{
int mid=l+(r-l)/3;int midmid=mid+(r-l)/3;
if(check(mid)>check(midmid))
l=mid;
else
r=midmid;
}
int res=2e18;
for(int i=l;i<=r;i++)
res=min(res,check(i));
cout<<res<<endl;
}