第61场牛客小白月赛-题解

第61场牛客小白月赛-题解

原题链接：C-小喵觅食

算法(枚举,BFS)

我们考虑猫$M$能够到达的点，并通过$BFS$求出猫$M$到达这些点的最短距离；由于猫$M$只会在属于自己的$r2$范围内闻到食物的味道，并且去尝试找到它，最后吃掉它，所以我们需要枚举小猫闻到哪个位置的食物（在$r2$范围外，小猫是无法闻到的，所以也就会待在原地），由于小美$P$只会在$r1$范围内移动，所以求解$P$能够到达的点的最短距离时，要保证当前点一定在范围内。

AC_Code

#include <iostream>
#include <cstring>
#include <algorithm>
#include <queue>

#define x first
#define y second

using namespace std;

const int N = 1010, INF = 0x3f3f3f3f;

typedef pair<int, int> PII;

int n, m, r1, r2;
char g[N][N];
int d1[N][N], d2[N][N];

int dx[4] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1};

void bfs_1(int sx, int sy)
{
    memset(d1, 0x3f, sizeof d1);
    d1[sx][sy] = 0;
    queue<PII> q;
    
    q.push({sx, sy});
    while(q.size())
    {
        auto t = q.front(); q.pop();
        int x = t.x, y = t.y;
        for(int i = 0; i < 4; i ++)
        {
            int a = x + dx[i], b = y + dy[i];
            if(a >= 1 && a <= n && b >= 1 && b <= m && g[a][b] != '*' && abs(a - sx) + abs(b - sy) <= r1)
            {
                if(d1[a][b] > d1[x][y] + 1)
                {
                    d1[a][b] = d1[x][y] + 1;
                    q.push({a, b});
                }
            }
        }
    }
}

void bfs_2(int sx, int sy)
{
    memset(d2, 0x3f, sizeof d2);
    d2[sx][sy] = 0;
    queue<PII> q;
    
    q.push({sx, sy});
    while(q.size())
    {
        auto t = q.front(); q.pop();
        int x = t.x, y = t.y;
        for(int i = 0; i < 4; i ++)
        {
            int a = x + dx[i], b = y + dy[i];
            if(a >= 1 && a <= n && b >= 1 && b <= m && g[a][b] != '*')
            {
                if(d2[a][b] > d2[x][y] + 1)
                {
                    d2[a][b] = d2[x][y] + 1;
                    q.push({a, b});
                }
            }
        }
    }
}

int main()
{
    scanf("%d%d%d%d", &n, &m, &r1, &r2);
    for(int i = 1; i <= n; i ++) scanf("%s", g[i] + 1);
    
    int x = 0, y = 0;
    for(int i = 1; i <= n; i ++)
        for(int j = 1; j <= m; j ++)
            if(g[i][j] == 'P') bfs_1(i, j);
            else if(g[i][j] == 'M') x = i, y = j, bfs_2(i, j); // 猫的位置
    
    int res = INF;
    for(int i = 1; i <= n; i ++)
        for(int j = 1; j <= m; j ++)
            if(abs(i - x) + abs(j - y) <= r2) // 猫能够到达的点
                res = min(res, d1[i][j] + d2[i][j]);

    printf("%d\n", res == INF ? -1 : res);
    return 0;
}