【CCF】201312-5 I’m stuck!（BFS）

试题编号： 201312-5
试题名称： I’m stuck!
时间限制： 1.0s
内存限制： 256.0MB
问题描述：
问题描述
　　给定一个R行C列的地图，地图的每一个方格可能是’#’, ‘+’, ‘-’, ‘|’, ‘.’, ‘S’, ‘T’七个字符中的一个，分别表示如下意思：
　　‘#’: 任何时候玩家都不能移动到此方格；
　　‘+’: 当玩家到达这一方格后，下一步可以向上下左右四个方向相邻的任意一个非’#‘方格移动一格；
　　‘-’: 当玩家到达这一方格后，下一步可以向左右两个方向相邻的一个非’#‘方格移动一格；
　　‘|’: 当玩家到达这一方格后，下一步可以向上下两个方向相邻的一个非’#‘方格移动一格；
　　‘.’: 当玩家到达这一方格后，下一步只能向下移动一格。如果下面相邻的方格为’#’，则玩家不能再移动；
　　‘S’: 玩家的初始位置，地图中只会有一个初始位置。玩家到达这一方格后，下一步可以向上下左右四个方向相邻的任意一个非’#‘方格移动一格；
　　‘T’: 玩家的目标位置，地图中只会有一个目标位置。玩家到达这一方格后，可以选择完成任务，也可以选择不完成任务继续移动。如果继续移动下一步可以向上下左右四个方向相邻的任意一个非’#‘方格移动一格。
　　此外，玩家不能移动出地图。
　　请找出满足下面两个性质的方格个数：
　　1. 玩家可以从初始位置移动到此方格；
　　2. 玩家不可以从此方格移动到目标位置。
输入格式
　　输入的第一行包括两个整数R 和C，分别表示地图的行和列数。(1 ≤ R, C ≤ 50)。
　　接下来的R行每行都包含C个字符。它们表示地图的格子。地图上恰好有一个’S’和一个’T’。
输出格式
　　如果玩家在初始位置就已经不能到达终点了，就输出“I’m stuck!”（不含双引号）。否则的话，输出满足性质的方格的个数。
样例输入
5 5
–±+
…|#.
…|##
S-±T
####.
样例输出
2
样例说明
　　如果把满足性质的方格在地图上用’X’标记出来的话，地图如下所示：
　　--±+
　　…|#X
　　…|##
　　S-±T
　　####X

搜索问题，根据题目描述搜索分两步：
**1.**找到初始位置能够到达的所有方格（如果到达不了目标位置输出“I’m stuck!”）
2. 统计 1. 中能到达的方格中所有到达不了目标位置的点的个数

这里我们采用BFS（广度优先搜索）的方法，注意数组大小
（总感觉这样暴力着写可能会T，但是这个题数据量才50*50，所以就暂时这么写了，感觉可以双向BFS，回头有空了再看看（逃））

#include <bits/stdc++.h>

using namespace std;
struct dir
{
    int x, y;
} d[4] = {{-1, 0}, {1, 0}, {0, -1}, {0, 1}};

typedef long long ll;
const int N = 50 + 5;
char mp[N][N];
int vis[N][N], vis2[N][N];
int n, m;
int ans = 0;
int bfs(int x, int y)
{
    vis[x][y] = 1;
    queue<pair<int, int>> q;
    q.push({x, y});
    while (!q.empty())
    {
        int xx = q.front().first;
        int yy = q.front().second;
        q.pop();
        if (mp[xx][yy] == 'S' || mp[xx][yy] == '+' || mp[xx][yy] == 'T')
        {
            for (int i = 0; i < 4; i++)
            {
                if (0 <= xx + d[i].x && xx + d[i].x < n && 0 <= yy + d[i].y && yy + d[i].y < m && !vis[xx + d[i].x][yy + d[i].y] && mp[xx + d[i].x][yy + d[i].y] != '#')
                    q.push({xx + d[i].x, yy + d[i].y}), vis[xx + d[i].x][yy + d[i].y] = 1;
            }
        }

        else if (mp[xx][yy] == '-')
        {
            for (int i = 2; i < 4; i++)
            {
                if (0 <= xx + d[i].x && xx + d[i].x < n && 0 <= yy + d[i].y && yy + d[i].y < m && !vis[xx + d[i].x][yy + d[i].y] && mp[xx + d[i].x][yy + d[i].y] != '#')
                    q.push({xx + d[i].x, yy + d[i].y}), vis[xx + d[i].x][yy + d[i].y] = 1;
            }
        }

        else if (mp[xx][yy] == '|')
        {
            for (int i = 0; i < 2; i++)
            {
                if (0 <= xx + d[i].x && xx + d[i].x < n && 0 <= yy + d[i].y && yy + d[i].y < m && !vis[xx + d[i].x][yy + d[i].y] && mp[xx + d[i].x][yy + d[i].y] != '#')
                    q.push({xx + d[i].x, yy + d[i].y}), vis[xx + d[i].x][yy + d[i].y] = 1;
            }
        }

        else if (mp[xx][yy] == '.')
        {

            if (0 <= xx + d[1].x && xx + d[1].x < n && 0 <= yy + d[1].y && yy + d[1].y < m && !vis[xx + d[1].x][yy + d[1].y] && mp[xx + d[1].x][yy + d[1].y] != '#')
                q.push({xx + d[1].x, yy + d[1].y}), vis[xx + d[1].x][yy + d[1].y] = 1;
        }
    }
}
int bfs1(int x, int y)
{
    vis2[x][y] = 1;
    queue<pair<int, int>> q;
    q.push({x, y});
    while (!q.empty())
    {
        int xx = q.front().first;
        int yy = q.front().second;
        q.pop();
        if (mp[xx][yy] == 'S' || mp[xx][yy] == '+' || mp[xx][yy] == 'T')
        {
            for (int i = 0; i < 4; i++)
            {
                if (0 <= xx + d[i].x && xx + d[i].x < n && 0 <= yy + d[i].y && yy + d[i].y < m && !vis2[xx + d[i].x][yy + d[i].y] && mp[xx + d[i].x][yy + d[i].y] != '#')
                    q.push({xx + d[i].x, yy + d[i].y}), vis2[xx + d[i].x][yy + d[i].y] = 1;
            }
        }

        else if (mp[xx][yy] == '-')
        {
            for (int i = 2; i < 4; i++)
            {
                if (0 <= xx + d[i].x && xx + d[i].x < n && 0 <= yy + d[i].y && yy + d[i].y < m && !vis2[xx + d[i].x][yy + d[i].y] && mp[xx + d[i].x][yy + d[i].y] != '#')
                    q.push({xx + d[i].x, yy + d[i].y}), vis2[xx + d[i].x][yy + d[i].y] = 1;
            }
        }

        else if (mp[xx][yy] == '|')
        {
            for (int i = 0; i < 2; i++)
            {
                if (0 <= xx + d[i].x && xx + d[i].x < n && 0 <= yy + d[i].y && yy + d[i].y < m && !vis2[xx + d[i].x][yy + d[i].y] && mp[xx + d[i].x][yy + d[i].y] != '#')
                    q.push({xx + d[i].x, yy + d[i].y}), vis2[xx + d[i].x][yy + d[i].y] = 1;
            }
        }

        else if (mp[xx][yy] == '.')
        {

            if (0 <= xx + d[1].x && xx + d[1].x < n && 0 <= yy + d[1].y && yy + d[1].y < m && !vis2[xx + d[1].x][yy + d[1].y] && mp[xx + d[1].x][yy + d[1].y] != '#')
                q.push({xx + d[1].x, yy + d[1].y}), vis2[xx + d[1].x][yy + d[1].y] = 1;
        }
    }
}
int main()
{
    int x1, y1, x2, y2;
    cin >> n >> m;
    for (int i = 0; i < n; i++)
        for (int j = 0; j < m; j++)
        {
            cin >> mp[i][j];
            if (mp[i][j] == 'S')
                x1 = i, y1 = j;
            else if (mp[i][j] == 'T')
                x2 = i, y2 = j;
        }
    bfs(x1, y1);
    if (vis[x2][y2])
    {
        int ans = 0;
        for (int i = 0; i < n; i++)
            for (int j = 0; j < m; j++)
            {
                if (vis[i][j])
                {
                    memset(vis2, 0, sizeof(vis2));
                    bfs1(i, j);

                    if (!vis2[x2][y2])
                        ans++;
                }
            }

        cout << ans << endl;
    }
    else
        puts("I'm stuck! ");

    return 0;
}
/*
5 5
--+-+
..|#.
..|##
S-+-T
####.
*/