P1141 01迷宮 (bfs)
阿新 • • 發佈：2018-12-06
題目描述
有一個僅由數字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。
題意：對於每次詢問（i,j）找到點（i,j）可以到達的格子個數（包括自身）
思路：
如果點（i1,j1）可以到達點(i2,j2),那麼點(i1,j1)可以到達其他點個數與點(i2,j2)可以到達其他點個數是相同的，這樣我們就不用對每個點bfs一遍(否則會超時)，我們只需要將點（i,j）所有可以到達的點標記在一起存入ans陣列中就好。
程式碼如下:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 |
#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<queue>
#include<math.h>
#define ll long long
using namespace std;
int n,m;
struct Node
{
int x;
int y;
};
int mp[1005][1005];
int ans[1000005]; // 答案陣列 開小會WA
int kis[1005][1005]; //標記陣列 如果kis[i][j]=-1,則表示該點未到達過
int fx[4]={0,0,1,-1}; // 對於每一個點可以到達的方向
int fy[4]={1,-1,0,0};
int k=0;
Node u,v;
queue<Node>q;
int bfs(int x,int y)
{
u.x=x;
u.y=y;
int sum=1;
q.push(u);
while(!q.empty())
{
v=q.front();
q.pop();
for(int i=0;i<4;i++)
{
int xx=v.x+fx[i];
int yy=v.y+fy[i];
if(xx>=1&&xx<=n&&yy>=1&&yy<=n&&kis[xx][yy]==-1&&(mp[xx][yy]==1-mp[v.x][v.y])) // 如果該點未到達過並且滿足條件
{
sum++;
kis[xx][yy]=k; // 將（xx,yy）點標記為k 所有kis[][]=k的點具有相同答案
u.x=xx;
u.y=yy;
//vis[xx][yy]=true;
q.push(u);
}
}
}
ans[k]=sum; // 將標記為K的點能到達其他點的個數存入ans陣列
}
int main()
{
scanf("%d%d",&n,&m);
char str[1005];
for(int i=0;i<n;i++)
{
scanf("%s",str);
for(int j=0;j<n;j++)
{
mp[i+1][j+1]=str[j]-'0';
}
}
memset(kis,-1,sizeof(kis));
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
if(kis[i][j]==-1) //如果該點未到達過，對該點進行bfs
{
kis[i][j]=k;
bfs(i,j);
k++;
}
}
}
int x,y;
for(int i=0;i<m;i++)
{
scanf("%d%d",&x,&y);
printf("%d\n",ans[kis[x][y]]);
}
return 0;
} |