利用二分法求解
題意:
??? 一個(gè)農(nóng)夫帶著k頭牛去住店,(人一間,每牛一間)已知該旅店共有n間房,其中部分房間已有人住,房間住宿情況由01串表示,0表示空,1表示已有人住,剩余房間足夠容納(k+1)頭牛/人。為了保障牛的安全,希望人住的房間離最遠(yuǎn)的牛的房間位置盡量小,輸出最小距離。
思路:
??? 很明顯為了讓人住的離最遠(yuǎn)的牛最近,那么最后這批人牛住的房間肯定是連續(xù)的(不算原有的住宿人員),且人住的位置離中心點(diǎn)越近越好。首先,枚舉住宿的左區(qū)間,如果左端點(diǎn)已有人住,則跳過該點(diǎn),如果空,則二分以該點(diǎn)為左端點(diǎn),空閑房間數(shù)為k+1的最左位置。隨后在這個(gè)區(qū)間內(nèi),開始尋找空閑的最中心位置(距離遠(yuǎn)的那端盡量?。脜^(qū)間長(zhǎng)度奇偶性設(shè)置兩個(gè)指針p1,p2的初始位置,隨后分別往兩端移動(dòng),因?yàn)橐欢ㄓ锌臻e位置,且兩指針同時(shí)移動(dòng),不會(huì)出現(xiàn)越界情況。最后找到一個(gè)解,則計(jì)算最遠(yuǎn)距離,若小于最優(yōu)值,則更新。
代碼:
#include#include#include#includeusing?namespace?std;
char?s[100010];
int?room[100010];
int?main()
{
????int?n,k,len,le,ri,border,ans,pos;
//讀入
scanf("%d%d",&n,&k);
scanf("%s",s);
room[0]=0;
for(int?i=1;i<=n;i++)
{
//前綴和
if(s[i-1]-'0')
room[i]=room[i-1];
else
room[i]=room[i-1]+1;
}
//設(shè)置一個(gè)肯定會(huì)被更新的最大值
ans=10e6;
for(int?i=1;i<=n;i++)
{
//該點(diǎn)已有人住
???if(room[i]==room[i-1])
???continue;
???????le=i;
???ri=n;
???border=-1;
???//二分右區(qū)間
???while(le>1;
???if(room[mid]-room[i-1]>k)
???{
???border=mid;
???ri=mid-1;
???}
???else
???le=mid+1;
???}
???//如果能找到k+1個(gè)房間
???if(border!=-1)
???{
?????????int?len=(border-i),p1,p2;
?//根據(jù)區(qū)間長(zhǎng)度奇偶性,設(shè)置p1,p2
?????????if(len%2)
?{
?p1=(border+i)>>1;
?p2=(border+i)/2+1;
?}
?else
?{
?p1=p2=(border+i)>>1;
?}
?//尋找最先出現(xiàn)空閑房間
?while(1)
?{
?if((room[p1]!=room[p1-1]))
????{
pos=p1;
break;
}
?else?if(room[p2]!=room[p2-1])
?{
?pos=p2;
?break;
}
?p1--;
?p2++;
?}
?????????//更新最優(yōu)值
?ans=min(ans,max(pos-i,border-pos));
???}
???//說明剩下,已無可能有k+1個(gè)房間
???else
???break;
}
printf("%dn",ans);
return?0;
}




