E. Restorer Distance(三分,2100)

题目传送门

题意 给你一个长度为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;
}

原创文章 144 获赞 13 访问量 8650

猜你喜欢

转载自blog.csdn.net/Joker_He/article/details/106171896