洛谷 P1141 01迷宫（普及/提高-）

思路：深搜，判断连通分支数
错误：初始化数组过大，超时（也只AC了5个点）
输入字符串的范围有误

题目描述

有一个仅由数字000与111组成的n×nn \times nn×n格迷宫。若你位于一格0上，那么你可以移动到相邻444格中的某一格111上，同样若你位于一格1上，那么你可以移动到相邻444格中的某一格000上。

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

第111行为两个正整数n,mn,mn,m。

下面nnn行，每行nnn个字符，字符只可能是000或者111，字符之间没有空格。

接下来mmm行，每行222个用空格分隔的正整数i,ji,ji,j，对应了迷宫中第iii行第jjj列的一个格子，询问从这一格开始能移动到多少格。

输出格式：

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

输入输出样例
输入样例#1： 复制

2 2
01
10
1 1
2 2

输出样例#1： 复制

4
4

说明

所有格子互相可达。

对于20%20%20%的数据，n≤10n≤10n≤10；

对于40%40%40%的数据，n≤50n≤50n≤50；

对于50%50%50%的数据，m≤5m≤5m≤5；

对于60%60%60%的数据，n≤100,m≤100n≤100,m≤100n≤100,m≤100；

对于100%100%100%的数据，n≤1000,m≤100000n≤1000,m≤100000n≤1000,m≤100000。

#include<iostream>
#include<cstdio>
#include<cstring>
int book[1002][1002]={0},n,s[1000005],max;
char a[1003][1003];
int dfs(int i,int j,int flag)
{
	if(book[i][j]!=0)
		return 0;	
	book[i][j]=flag;
	max++;	
//	printf("%d\n",max);
	if(a[i][j]=='1')
	{	
//		printf("dfs1\n");
		if(a[i-1][j]=='0'&&i-1>=0)		dfs(i-1,j,flag);	
		if(a[i+1][j]=='0'&&i+1<n)		dfs(i+1,j,flag);
		if(a[i][j-1]=='0'&&j-1>=0)		dfs(i,j-1,flag);	
		if(a[i][j+1]=='0'&&j+1<n)		dfs(i,j+1,flag);
	}
	else if(a[i][j]=='0')
	{	
//		printf("dfs0\n");
		if(a[i-1][j]=='1'&&i-1>=0)		dfs(i-1,j,flag);	
		if(a[i+1][j]=='1'&&i+1<n)		dfs(i+1,j,flag);
		if(a[i][j-1]=='1'&&j-1>=0)		dfs(i,j-1,flag);	
		if(a[i][j+1]=='1'&&j+1<n)		dfs(i,j+1,flag);
	}
	return max;
}
int main()
{
	freopen("P1141.in","r",stdin);
	int m,i,j,p,q;
	scanf("%d%d",&n,&m);
	for(i=0;i<n;i++)
	{
		scanf("%s",&a[i]);
	}
//	printf("%c\n",a[0][0]);
	int flag=2;
	for(i=0;i<n;i++)
	{
		for(j=0;j<n;j++)
		{
			max=0;
			s[flag]=dfs(i,j,flag);
			if(s[flag]!=0)
				flag++;
		}
	}
//	for(i=2;i<=flag;i++)
//		sa=s[i]+sa;
//	printf("s=====%d\n",sa);
	for(i=0;i<m;i++)
	{
		scanf("%d%d",&p,&q);
		if(p<=n&&q<=n)
			printf("%d\n",s[book[p-1][q-1]]);
	}
	
	return 0;
 }