P1141 01迷宫

P1141 01迷宫
题目描述

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

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

输入输出格式

输入格式：
输入的第1行为两个正整数n，m。

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

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

输出格式：
输出包括m行，对于每个询问输出相应答案。

输入输出样例

输入样例#1：
2 2
01
10
1 1
2 2
输出样例#1：
4
4
说明

所有格子互相可达。

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

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

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

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

写了下面一段代码，但是在https://www.luogu.org/problem/show?pid=1141#sub测试一下，只有70分，其他的三个测试点提示超时，代码如下：

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
#define N 1010
#define M 100010

int a[N][N], t[N][N], b[M][2], xx[]={0,0,-1,1}, yy[]={-1,1,0,0}, n, m, ans;
void f(int x, int y)
{
	ans++;
	if(a[x][y] == 0)
	{
		a[x][y] = -1;
		for(int i=0; i<4; i++)
		{
			if(a[x+xx[i]][y+yy[i]] == 1)
				f(x+xx[i], y+yy[i]);
		}
	//	a[x][y] = 0;
	}
	if(a[x][y] == 1)
	{
		a[x][y] = -1;
		for(int i=0; i<4; i++)
		{
			if(a[x+xx[i]][y+yy[i]] == 0)
				f(x+xx[i], y+yy[i]);
		}
	//	a[x][y] = 1;
	}
}

int main()
{
	int i, j;
	cin>>n>>m;
	memset(t, -1, sizeof(t));
	for(i=1; i<=n; i++)
	{
		for(j=1; j<=n; j++)
			scanf("%1d", &t[i][j]);
	}
	for(i=1; i<=m; i++)
		cin>>b[i][0]>>b[i][1];
	
	for(i=1; i<=m; i++)
	{
		memcpy(a, t, sizeof(t));
		ans = 0;
		f(b[i][0], b[i][1]);
		cout<<ans<<endl;
	}
		
	return 0;
}

我对上面的代码进行了优化，把所有cin和cout换成了scanf和printf，这个两个函数输入输出快一点，然后对递归函数进行了优化，连很少使用的异或也用进来了，也取消了一些重复计算，比如循环里面的sizeof，代码如下：

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
#define N 1010
#define M 100010

int a[N][N], t[N][N]={0}, b[M][2], xx[]={0,0,-1,1}, yy[]={-1,1,0,0}, n, m, ans, nx, ny;

void f(int x, int y)
{
    ans++;
    t[x][y] = 1;
    for(int i=0; i<4; i++)
    {
        nx = x+xx[i];
        ny = y+yy[i];
        if(nx>=1 && ny>=1 && nx<=n && ny<=n && !t[nx][ny] && (a[x][y]^a[nx][ny]))
            f(nx, ny);
    }
//    t[x][y] = 0;
}

int main()
{
    int i, j;
    cin>>n>>m;
 //   memset(a, -1, sizeof(a));
    for(i=1; i<=n; i++)
    {
        for(j=1; j<=n; j++)
            scanf("%1d", &a[i][j]);
    }
    for(i=1; i<=m; i++)
        scanf("%d %d", &b[i][0], &b[i][1]);
    
    for(int size=sizeof(t),i=1; i<=m; i++)
    {
        memset(t, 0, size); //相应位置为0表示这个点没有去过 
        ans = 0;
        f(b[i][0], b[i][1]);
        printf("%d\n", ans);
    }
        
    return 0;
}

然并卵，我就算把main里面对主计算函数屏蔽掉也还是提示超时，头大了，可能要对程序实现进一步改动才行。

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
#define N 1010
#define M 100010

int a[N][N], t[N][N]={0}, b[M][2], xx[]={0,0,-1,1}, yy[]={-1,1,0,0}, n, m, ans, nx, ny;
int aa[N][N], saveans[M], k;

void f(int x, int y)
{
    ans++;
    aa[x][y] = k;
    t[x][y] = 1;
    for(int i=0; i<4; i++)
    {
        nx = x+xx[i];
        ny = y+yy[i];
        if(nx>=1 && ny>=1 && nx<=n && ny<=n && !t[nx][ny] && (a[x][y]^a[nx][ny]))
            f(nx, ny);
    }
//    t[x][y] = 0;
}

int main()
{
    int i, j;
    cin>>n>>m;
 //   memset(a, -1, sizeof(a));
    for(i=1; i<=n; i++)
    {
        for(j=1; j<=n; j++)
            scanf("%1d", &a[i][j]);
    }
    for(i=1; i<=m; i++)
        scanf("%d %d", &b[i][0], &b[i][1]);
        
    int size=sizeof(t);
    memset(aa, -1, size);
    memset(t, 0, size); //相应位置为0表示这个点没有去过
    k = 0;
    for(i=1; i<=m; i++)
    {
        if(aa[b[i][0]][b[i][1]] == -1)
        {
            k++; 
            ans = 0;
            f(b[i][0], b[i][1]);
            saveans[k] = ans;
        }      
        printf("%d\n", saveans[aa[b[i][0]][b[i][1]]]);
    }
        
    return 0;
}

终于圆满解决问题。