洛谷 - P1141 - 01迷宫 - dfs

https://www.luogu.org/problemnew/show/P1141

能互相到达的格子的答案自然是一样的，第一次dfs标记联通块，第二次dfs把cnt传递到整个联通卡并顺手消除vis标记（其实把vis标记改成另一个也可以的）。

#include<bits/stdc++.h>
using namespace std;
#define ll long long

int n,m;
int g[][];
int ans[][];

int cnt=;
int vis[][];

void dfs(int r,int c){
    //printf("%d %d\n",r,c);
    vis[r][c]=;
    cnt++;
    if(r->=&&vis[r-][c]==&&g[r][c]!=g[r-][c])
        dfs(r-,c);
    if(r+<=n&&vis[r+][c]==&&g[r][c]!=g[r+][c])
        dfs(r+,c);
    if(c->=&&vis[r][c-]==&&g[r][c]!=g[r][c-])
        dfs(r,c-);
    if(c+<=n&&vis[r][c+]==&&g[r][c]!=g[r][c+])
        dfs(r,c+);
}

void dfs2(int r,int c){
    vis[r][c]=;
    ans[r][c]=cnt;
    if(r->=&&vis[r-][c])
        dfs2(r-,c);
    if(r+<=n&&vis[r+][c])
        dfs2(r+,c);
    if(c->=&&vis[r][c-])
        dfs2(r,c-);
    if(c+<=n&&vis[r][c+])
        dfs2(r,c+);
}

int main(){
    scanf("%d%d",&n,&m);
    for(int i=;i<=n;i++){
        for(int j=;j<=n;j++){
            scanf("%1d",&g[i][j]);
            //printf("%d!\n",g[i][j]);
        }
    }

    for(int i=;i<=n;i++){
        for(int j=;j<=n;j++){
            if(ans[i][j]==){
                cnt=;
                //memset(vis,0,sizeof(vis));
                dfs(i,j);
                dfs2(i,j);
            }
        }
    }

    for(int i=;i<=m;i++){
        int r,c;
        scanf("%d%d",&r,&c);
        printf("%d\n",ans[r][c]);
    }
}