【ACM- OJ】《01迷宫》C++

题目描述

有一个仅由数字0与1组成的n×n格迷宫。若你位于一格0上，那么你可以移动到相邻4格中的某一格1上，同样若你位于一格1上，那么你可以移动到相邻4格中的某一格0上。 你的任务是：对于给定的迷宫，询问从某一格开始能移动到多少个格子（包含自身）。

输入

第1行为两个正整数n,m。 下面n行，每行n个字符，字符只可能是0或者1，字符之间没有空格。
接下来m行，每行2个用空格分隔的正整数i,j，对应了迷宫中第i行第j列的一个格子，询问从这一格开始能移动到多少格。

输出

m行，对于每个询问输出相应答案。

样例输入

2 2
01
10
1 1
2 2

样例输出

4
4

提示

n <= 400

AC代码

//
//  main.cpp
//  01迷宫
//
//  Created by HISS on 2020/11/5.
//  Copyright © 2020 HISS. All rights reserved.
//

#include <iostream>
using namespace std;

int n, m, ans;
char map[405][405];
int visited[405][405];

bool canbe(int x, int y)
{
    if(x < 1 || x > n || y < 1 || y > n || visited[x][y] == 1)
        return false;
    else
        return true;
}

void dfs(int x, int y)
{
    if(!canbe(x, y))
        return;
    visited[x][y] = 1;
    if(map[x][y + 1] != map[x][y] && canbe(x, y + 1))
    {
        ans ++;
        dfs(x, y + 1);
    }
    if(map[x][y - 1] != map[x][y] && canbe(x, y - 1))
    {
        ans ++;
        dfs(x, y - 1);
    }
    if(map[x - 1][y] != map[x][y] && canbe(x - 1, y))
    {
        ans ++;
        dfs(x - 1, y);
    }
    if(map[x + 1][y] != map[x][y] && canbe(x + 1, y))
    {
        ans ++;
        dfs(x + 1, y);
    }
    return;
}

int main() {
    cin >> n >> m;
    for(int i = 1; i <= n ; i ++)
        for(int j = 1; j <= n; j ++)
            cin >> map[i][j];
    while(m --)
    {
        int x, y;
        cin >> x >> y;
        for(int i = 1; i <= n; i ++)
            for(int j = 1; j <= n; j ++)
                visited[i][j] = 0;
        ans = 1;
        dfs(x, y);
        cout << ans << endl;
    }
    return 0;
}