HDU-2612 Find a way（双向bfs）

Find a way

题意：给你一维图，两个人Y，M，约在kfc(@)见面,求两个人走的步数和最小的走法

做法:分别从两个人的位置进行bfs，将两次的结果写入ans数组，最后遍历ans数组，找出两个人都能到且步数和最小的kfc。

代码：

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
const int N = 200 + 10;
const int INF = 0x3f3f3f3f;

char a[N][N];
int vis[N][N];
int ans[2][N][N];
int dir[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
int r, c, x1, yy, x2, y2;
struct node{
    int x, y, step;
    node(int x, int y, int step):x(x),y(y),step(step){}
};

void result()
{
    int cnt = INF;
    // for (int i = 0; i < r; i++){
    //     for(int j = 0; j < c; j++){
    //         printf("%d ", ans[0][i][j]);
    //     }
    //     cout << endl;
    // }
    // cout << endl;
    // for (int i = 0; i < r; i++){
    //     for(int j = 0; j < c; j++){
    //         printf("%d ", ans[1][i][j]);
    //     }
    //     cout << endl;
    // }
    for (int i = 0; i < r; i++){
        for(int j = 0; j < c; j++){
            if(ans[0][i][j] && ans[1][i][j]){		//当两个人都能到达此处时，记录下来后面比较
                cnt = min(cnt, ans[0][i][j] + ans[1][i][j]);
            }
        }
    }
    printf("%d\n", cnt * 11);
}
bool f(int x, int y)
{
    if (x >=0 && x < r && y >= 0 && y < c){
        return true;
    }
    return false;
}
void bfs(int x, int y, int flag)
{
    memset(vis, 0, sizeof(vis));
    queue<node> s;
    s.push(node(x, y, 0));
    vis[x][y] = 1;
    while(!s.empty()){
        node cnt = s.front();
        s.pop();
        for (int i = 0; i < 4; i++){
            int tx = cnt.x + dir[i][0];
            int ty = cnt.y + dir[i][1];
            if (f(tx, ty) && !vis[tx][ty] && a[tx][ty] != '#'){  //注意：只要不是#的地方都能走，也就是说Y可以走M， @上。
                vis[tx][ty] = 1;
                s.push(node(tx, ty, cnt.step + 1));
                if (a[tx][ty] == '@')
                    ans[flag][tx][ty] = cnt.step + 1;
            }
        }
    }
}
int main()
{
    while(cin >> r >> c){
        memset(ans, 0, sizeof(ans));
        for (int i = 0; i < r; i++){
            for (int j = 0; j < c; j++){
                cin >> a[i][j];
                if (a[i][j] == 'Y'){
                    x1 = i;
                    yy = j;
                }
                if (a[i][j] == 'M'){
                    x2 = i;
                    y2 = j;
                }
            }
        }
        bfs(x1, yy, 0);		//分别进行bfs遍历
        bfs(x2, y2, 1);
        result();
    }
    return 0;
}
// freopen("in.txt", "r", stdin);
// freopen("out.txt", "w", stdout);