1. 程式人生 > >洛谷P1141 01迷宮【DFS】

洛谷P1141 01迷宮【DFS】

有一個僅由數字00與11組成的n \times nn×n格迷宮。若你位於一格0上,那麼你可以移動到相鄰44格中的某一格11上,同樣若你位於一格1上,那麼你可以移動到相鄰44格中的某一格00上。

你的任務是:對於給定的迷宮,詢問從某一格開始能移動到多少個格子(包含自身)。

輸入輸出格式

輸入格式:

第11行為兩個正整數n,mn,m。

下面nn行,每行nn個字元,字元只可能是00或者11,字元之間沒有空格。

接下來mm行,每行22個用空格分隔的正整數i,ji,j,對應了迷宮中第ii行第jj列的一個格子,詢問從這一格開始能移動到多少格。

輸出格式:

mm行,對於每個詢問輸出相應答案。

輸入輸出樣例

輸入樣例#1: 複製

2 2
01
10
1 1
2 2

輸出樣例#1: 複製

4
4

說明

所有格子互相可達。

對於20\%20%的資料,n≤10n≤10;

對於40\%40%的資料,n≤50n≤50;

對於50\%50%的資料,m≤5m≤5;

對於60\%60%的資料,n≤100,m≤100n≤100,m≤100;

對於100\%100%的資料,n≤1000,m≤100000n≤1000,m≤100000。

#include<cstdio>
#include<cstring>
int n,m,ans[100002],x,y,f[1002][1002];
char s[1002][1002];
void dfs(int r,int c,int z,int lll){
    if (r<0 || r>=n || c<0 || c>=n || f[r][c]!=-1 || s[r][c]-'0'!=z)return;
    f[r][c]=lll;ans[lll]++;
    dfs(r-1,c,!z,lll);dfs(r+1,c,!z,lll);dfs(r,c-1,!z,lll);dfs(r,c+1,!z,lll);
}
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=0;i<n;i++)
    scanf("%s",s[i]);
    memset(f,-1,sizeof(f));
    for (int i=0;i<m;i++)
    {
        scanf("%d%d",&x,&y);x--;y--;
        if (f[x][y]==-1)dfs(x,y,s[x][y]-'0',i);else ans[i]=ans[f[x][y]];
    }
    for (int i=0;i<m;i++)
    printf("%d\n",ans[i]);
    return 0;
}