Vasya and Robot(二分查詢)
阿新 • • 發佈:2018-12-20
-
題目傳送:
-
題意:
給出一段錯誤路徑,現要求修改路徑,使得能夠成功到達目標座標。求最小修改量。
-
思路:
若最終能使改動後到達目標座標(tarx,tary),設改動區間長度len,記刪除len內容後的位置座標為(nowx,nowy),計算出(nowx,nowy)與目標座標(tarx,tary)的曼哈頓距離dis。由於,行走時可來回反覆,只需滿足條件:修改的區間長度len與dis的奇偶性一致,且len大於等於dis即可滿足。題意要求最小改動區間長度,可以用二分法查詢符合條件的len的情況。
-
程式碼:
#include <iostream> #include <string> #define MAX 200005 using namespace std; int x[MAX]; int y[MAX]; bool judge(int n,int len,int tarx,int tary) { for (int i=1;i<=n-len+1;i++) { int nowx=x[i-1]+x[n]-x[i+len-1]; //求出刪去len段後的位置 int nowy=y[i-1]+y[n]-y[i+len-1]; int dis=abs(nowx-tarx)+abs(nowy-tary); //求曼哈頓距離 if ((len-dis)%2==0&&len>=dis) //判斷條件 return true; } return false; } int main() { int n; cin>>n; string str; cin>>str; int tarx,tary; cin>>tarx>>tary; for (int i=1;i<=n;i++) { if (str[i-1]=='R') x[i]=1; else if (str[i-1]=='L') x[i]=-1; else if (str[i-1]=='U') y[i]=1; else if (str[i-1]=='D') y[i]=-1; } for (int i=1;i<=n;i++) { x[i]=x[i]+x[i-1]; //記錄每一步完成後橫縱座標當前到達的位置 y[i]=y[i]+y[i-1]; } int sid=0,eid=n; int ans=-1,halfid; while(sid<=eid) //二分 { halfid=(sid+eid)/2; if(judge(n,halfid,tarx,tary)) { ans=halfid; eid=halfid-1; } else { sid=halfid+1; } } cout<<ans<<endl; }