洛谷P1141 01迷宫

题目链接：https://www.luogu.com.cn/problem/P1141
问题重述：
有一个仅由数字0与1组成的n×n格迷宫，你每次可以移动到与自身不一样的格子上，问你每个连通块有多少格。
思考：
搜索题，找连通块以及记录大小，可以用bfs或者dfs来做，但是要注意dfs要用记忆化搜索，否则会超时。（流下了不会记忆化的泪水QAQ）

两种写法对比：

可以看到dfs更快，代码量更少，但是bfs思考起来更简单，两种方法各有利弊。

bfs写法：
用队列广搜，边搜边用动态数组记录块的信息，搜完遍历将结果赋给动态数组内位置值。

#include <bits/stdc++.h>

using namespace std;
char a[1005][1005];
int b[1005][1005];
int c[1005][1005];
int x2[4]={0,0,1,-1};
int y2[4]={1,-1,0,0};
struct sa{
int x,y;
};
int n,m;
void bfs(int dx,int dy)
{
    int cnt=1;
    b[dx][dy]=1;
    sa tmp;
    queue<sa>q;
    vector<sa>p;
    q.push({dx,dy});
    while(!q.empty())
    {
        tmp=q.front();
        p.push_back(tmp);//动态数组存
        q.pop();
        int x1=tmp.x;
        int y1=tmp.y;
         for(int i=0;i<4;i++)
        {
            int x=x1+x2[i];
            int y=y1+y2[i];
            if(x>=1&&x<=n&&y>=1&&y<=n&&a[x][y]!=a[x1][y1]&&b[x][y]==0)
            {
                q.push({x,y});
                cnt++;
                b[x][y]=1;
            }
        }
    }
    vector<sa>::iterator it;
    for(it=p.begin();it!=p.end();++it)
    {
        sa pos=*it;
        c[pos.x][pos.y]=cnt;
    }
}

int main()
{
    int dx,dy;
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        cin>>a[i]+1;
    while(m--)
    {
        scanf("%d%d",&dx,&dy);
        if(b[dx][dy]==0)
        {
            bfs(dx,dy);
            printf("%d\n",c[dx][dy]);
        }
        else
        printf("%d\n",c[dx][dy]);
    }
    return 0;
}

dfs写法：
用f数组标记连通块个数，cnt记录是第几种连通块，vis记录是否访问过该位置。
注意ans数组一定要大！大！大！（流下了re的泪水）。
1000*1000开1e6合适。

#include <bits/stdc++.h>

using namespace std;
const int N=1100;
char a[N][N];
int vis[N][N],f[N][N];
int x2[5]={0,0,0,1,-1};
int y2[5]={0,-1,1,0,0};
int ans[1000005],n,m,cnt,p;
void dfs(int x,int y)
{
    f[x][y]=cnt;
    ans[f[x][y]]++;
    vis[x][y]=1;
    for(int i=1;i<=4;i++)
    {
        int x3=x+x2[i];
        int y3=y+y2[i];
        if(a[x3][y3]!=a[x][y]&&x3>=1&&x3<=n&&y3>=1&&y3<=n&&vis[x3][y3]==0)
        {
            vis[x3][y3]=1;
            dfs(x3,y3);
        }
    }
}

int main()
{
    //std::ios::sync_with_stdio(0);
    //freopen("2.in","r",stdin);
    //freopen("out.txt","w",stdout);
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            cin>>a[i][j];
        }
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            if(!vis[i][j])
            {
                cnt++;//若未访问过则是新的连通块，cnt++
                dfs(i,j);
            }
        }
    }
    int x,y;
    for(int i=1;i<=m;i++)
    {
        scanf("%d %d",&x,&y);
        printf("%d\n",ans[f[x][y]]);
    }
    return 0;
}

最近在复习搜索，发现啥也不会了T A T