Find a way——bfs

Find a way

题意:两个起点Y和M,到达同一个@,所以耗时累加。问两人到达同一个@用时最短是多少,前进一格11秒。
思路:for循环找到Y,M的位置,以此为起点进行bfs,然后再找到所有@位置的时间(两个人的时间已加好)的最小值。
代码

#include
#include
#include
#include
#include
#include
#include
using namespace std;
struct E
{
    int x,y,step;
} t,cur;
int  n,m;
int cost[1005][1005];
int vis[1001][1001];
char a[1005][1005];
int dir[4][2]= {{1,0},{0,1},{-1,0},{0,-1}};
int flag,ans;
void bfs(int i,int j)
{
    queue<E>q;
    cur.x=i;
    cur.y=j;
    cur.step=0;
    vis[cur.x][cur.y]=1;
    q.push(cur);
    while(!q.empty())
    {
        t=q.front();
        q.pop();
        cur.step=t.step+1;
        for(int i=0; i<4; i++)
        {
            cur.x=t.x+dir[i][0];
            cur.y=t.y+dir[i][1];
            if(cur.x>=0&&cur.x<n&&cur.y>=0&&cur.y<m&&!vis[cur.x][cur.y]&&a[cur.x][cur.y]!='#')
            {
                vis[cur.x][cur.y]=1;
                cost[cur.x][cur.y]+=cur.step;
                q.push(cur);
            }
        }
    }
}
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        memset(vis,0,sizeof(vis));
        memset(cost,0,sizeof(cost));
        flag=0;
        ans=0x3f3f3f3f;
        for(int i=0; i<n; i++)
        {
            for(int j=0; j<m; j++)
            {
                cin>>a[i][j];
            }
        }
        for(int i=0; i<n; i++)
        {
            for(int j=0; j<m; j++)
            {
                if(a[i][j]=='Y'||a[i][j]=='M')
                {
                    memset(vis,0,sizeof(vis));
                    bfs(i,j);
                }
            }
        }
        for(int i=0; i<n; i++)
        {
            for(int j=0; j<m; j++)
            {
                if(cost[i][j]!=0&&a[i][j]=='@'&&cost[i][j]<=ans)
                {
                    ans=cost[i][j];
                }
            }
        }
        printf("%d\n",ans*11);
    }
    return 0;
}

你可能感兴趣的:(bfs)