求助,BFS:5*TLE+2*MLE+3*WA

P1649 [USACO07OCT] Obstacle Course S

_mxy_ @ 2023-08-16 23:30:03

错的很惨 ,代码如下,BFS模板+转向判断

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=110;

struct node
{
    int x,y,to;//↑1 ↓2 ←3 →4
};
char ch[N][N];
int n;
int dx[4]={0,0,1,-1};
int dy[4]={1,-1,0,0};
int sum=0,ans=-1;
int ex,ey;

int bfs(int tow)
{
    queue<node> q;
    q.push({1,1,tow});
    while(!q.empty())
    {
        node t=q.front();
        q.pop();
        if(t.x==ex && t.y==ey)
        {
            return sum;
        }
        for(int i=0;i<4;i++)
        {
            int nx=t.x+dx[i];
            int ny=t.y+dy[i];
            int nto;
            if(nx>=1 && nx<=n && ny>=1 && ny<=n && ch[nx][ny]!='x')
            {
                if(t.to==1 || t.to==2)
                {
                    if(dx[i]==1)
                    {
                        sum++;
                        nto=4;
                    }
                    else if(dx[i]==-1)
                    {
                        sum++;
                        nto=3;
                    }
                    else if(dy[i]==1)
                    {
                        nto=1;
                    }
                    else
                    {
                        nto=2;
                    }
                }
                if(t.to==3 || t.to==4)
                {
                    if(dx[i]==1)
                    {
                        nto=4;
                    }
                    else if(dx[i]==-1)
                    {
                        nto=3;
                    }
                    else if(dy[i]==1)
                    {
                        sum++;
                        nto=1;
                    }
                    else
                    {
                        sum++;
                        nto=2;
                    }
                }
                q.push({nx,ny,nto});
            }
        }
    }
    return -1;
}

int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            cin>>ch[i][j];
            if(ch[i][j]=='B')
            {
                ex=i;
                ey=j;
            }
        }
    }
    for(int i=1;i<=4;i++)
    {
        sum=0;
        int x=bfs(i);
        if(x!=-1 && (x<ans || ans==-1))
        {
            ans=x;
        }
    }
    cout<<ans<<endl;
    return 0;
}

by _mxy_ @ 2023-08-16 23:42:21

加了剪枝以后 ,成功把两个TLE变成了MLE(捂脸)


|