2022牛客寒假算法基础集训营5疫苗小孩题解报告

题解报告

标签：二分

题目链接

来源：牛客网

解题思路：

考虑题目给出的n的数据范围，这道题要么二分要么动态规划
1针和0针的提升都为0，所以只考虑两针以上的情况，枚举第二针所在位置天数a[i],然后找到离a[i]+k和a[i]-k最近的两天计算最大贡献（2针或3针，打针有可能会降低手速）
虽然比赛时就考虑这种做法，但对二分理解不行一直写不对
比如查找a[i]-k
首先通过二分查找在1~i-1范围内第一个大于等于目标的值的位置pos，此处要分情况处理
找不到时 pos=i，两针不能同一天打不合理 将pos=i-1
找到了 当pos==1 显然不存在小于a[i]-k的值
当i>pos>1 需要比较两侧谁更接近a[i]-k
同理可得查找第三针的方法

代码实现：

#include <bits/stdc++.h>
using namespace std;
int a[1000500];

int main(){
    int n,k,w,q,ans,cnt;
    cnt=ans = 0;
    string s;
    cin>>n;
    cin>>s;
    cin>>k>>w>>q;
    for(int i=0;i<n;i++){
		if(s[i]=='0'){
			cnt++;
			a[cnt]=i;
		}
	}

	for(int i=2;i<cnt;i++){
		int pos1=lower_bound(a+1,a+i-1,a[i]-k)-a;
		int pos2=lower_bound(a+i+1,a+cnt,a[i]+k)-a;
		
		if(pos1==i) pos1--;
		else{
			if(pos1!=1&&(a[i]-k)-a[pos1-1]<a[pos1]-(a[i]-k)) pos1--;
		} 
		
		if(pos2>cnt) pos2=cnt;
		else{
			if(pos2!=i+1&&(a[i]+k)-a[pos2-1]<a[pos2]-(a[i]+k)) pos2--;
		}
		
		ans = max(ans,w-q*abs(k-abs(a[i]-a[pos1])));
		ans = max(ans,w-q*abs(k-abs(a[i]-a[pos2])));
		ans = max(ans,w-q*abs(k-abs(a[i]-a[pos1]))+w-q*abs(k-abs(a[i]-a[pos2])));
	}     
	cout<<ans<<endl;
}