1. 程式人生 > >Vasya and Robot(二分查詢)

Vasya and Robot(二分查詢)

  • 題目傳送:

  • 題意:

給出一段錯誤路徑,現要求修改路徑,使得能夠成功到達目標座標。求最小修改量。

  •  思路:

若最終能使改動後到達目標座標(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;
}