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

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

原题链接:C-小喵觅食

算法(枚举,BFS)

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

AC_Code

#include 
#include 
#include 
#include 

#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;
}

你可能感兴趣的:(牛客小白月赛,C++,算法,ACM,牛客小白月赛,C++,算法)