传送门:http://codeforces.com/problemset/problem/613/B
Tips:upper_bound返回的是元素值大于查找值的第一个元素的位置,如果插入元素大于数组中的全部元素啊,则返回值是last,注意此时的下标越界
如:number 序列 1,2,2,4,upper_bound(2)后返回的是3,即4的位置
lower_bound返回的是大于等于val的第一个元素的位置
本题目中skill数组中的元素值肯定是单调递增的,故用upper_bound
本题到自己打的时候真的是漏洞百出,好多细节的地方都没有注意到,在代码里面全部都加上了小注释
1.下标的选取真的很重要,由于本题需要枚举全部为max,即i为0和n的情况,所以一定要注意越界的问题,再加上upper_boud的使用,需要下标–,所以当取到最小的时候一定要特判,否则下标变为-1就越界了,连样例都过不了
2.预处理的细节一定要去想清楚了,到底是减掉多少个skill,应该是前面的i-1个!!
3.最主要的还是说下标的问题,应该随时注意越界和取得数量多少的问题,比如anspos第一个循环就应该是<,因为是有i个满血的,所以下标应该为0到anpos-1!!
4.还有就是for循环迭代的时候注意变量的增加时间,判断应该在什么时候增加,优美的写法可以把它加到与i++同步,当然这只是小tips!
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define pi pair<ll,ll>
#define fi first //skill
#define se second //id
const int maxn=1e5+10;
pi skill[maxn];
int n,cf,cm;ll m,a;
ll reachlow[maxn];
ll ans[maxn];
//downpos is 1,need test
//calculate the minval with remain money and remain num
ll solve(ll num,ll money){
if(num==0) return a;// be warn of the overflow of -1
ll pos=upper_bound(reachlow,reachlow+num,money)-reachlow;
pos--;
money-=reachlow[pos];
return min(skill[pos].fi+money/(pos+1),a);
}
ll maxx,anspos,minval,minnval;
int main() {
cin>>n>>a>>cf>>cm>>m;
for(int i=0;i<n;i++){
scanf("%d",&skill[i].fi);skill[i].se=i;
}
sort(skill,skill+n);
ll sum=0;
//this point downpos i needs sub 1
for(int i=0;i<n;i++){
reachlow[i]=skill[i].fi*i-sum;
sum+=skill[i].fi;
}
ll need=0;
//similariy i starts from 0 means all reach a
//the reason of max() is that when i==n
// i means the number of max && sub1
maxx=0;
for(int i=0;i<=n;need+=a-skill[max(n-i-1,0)].fi,i++){
//max is a
if(need>m) break;
minval=solve(n-i,m-need);
ll tmp=minval*cm+i*cf;
if(tmp>maxx){
maxx=tmp;anspos=i;minnval=minval;
}
}
// calculate ans
//be warn of the number of skill of max
for(int i=0;i<anspos;i++){
ans[skill[n-i-1].se]=a;
}
for(int i=anspos;i<n;i++){
ans[skill[n-i-1].se]=max(skill[n-i-1].fi,minnval);
}
printf("%lld\n",maxx);
for(int i=0;i<n;i++){
printf("%lld ",ans[i]);
}
printf("\n");
return 0;
}