01迷宫简易版

题目描述
​ 有一个仅由数字 0 与 1 组成的 n×m 格迷宫。若你位于一格 0 上，那么你可以移动到相邻 4 格中的某一格 1 上，同样若你位于一格 1 上，那么你可以移动到相邻 4 格中的某一格 0 上。

​ 你的任务是：对于给定的迷宫，询问从某一格开始能移动到多少个格子（包含自身）。

输入
​ 第 1 行为两个正整数 n,m。

​ 下面 n 行，每行 m 个字符，字符只可能是 0 或者 1，字符之间没有空格。

​ 接下来一行给出一个点的在坐标。

输出
​ 一个整数表示该点能移动到的格子数（包括自身）。

样例输入
2 3
011
100
2 3
样例输出
2
数据规模与约定
​ 时间限制：1 s

​ 内存限制：256 M

​ 100% 的数据保证 1≤n,m≤3000

#include<iostream>
using namespace std;

int n, m;
char map[3005][3005];
int dir[4][2] = { 1,0,0,1,-1,0,0,-1 };
int mem[3005][3005];
int cnt = 0;
void func(int x, int y) {
	for (int i = 0; i < 4; i++) {
		int x2 = x + dir[i][0];
		int y2 = y + dir[i][1];
		if (map[x2][y2] != 0 && map[x][y] != map[x2][y2] && mem[x2][y2] == 0){
			cnt++;
			mem[x2][y2] = 1;
			func(x2, y2);
		}
	}
}
int main() {
	cin >> n >> m;
	for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= m; j++) {
			cin >> map[i][j];
		}
	}
	int x, y;
	cin >> x >> y;
	mem[x][y] = 1;
	cnt = 1;
	func(x, y);
	cout << cnt << endl;
	return 0;
}