luogu P1332 血色先锋队[bfs]

题目描述

巫妖王的天灾军团终于卷土重来，血色十字军组织了一支先锋军前往诺森德大陆对抗天灾军团，以及一切沾有亡灵气息的生物。孤立于联盟和部落的血色先锋军很快就遭到了天灾军团的重重包围，现在他们将主力只好聚集了起来，以抵抗天灾军团的围剿。可怕的是，他们之中有人感染上了亡灵瘟疫，如果不设法阻止瘟疫的扩散，很快就会遭到灭顶之灾。大领主阿比迪斯已经开始调查瘟疫的源头。原来是血色先锋军的内部出现了叛徒，这个叛徒已经投靠了天灾军团，想要将整个血色先锋军全部转化为天灾军团！无需惊讶，你就是那个叛徒。在你的行踪败露之前，要尽快完成巫妖王交给你的任务。

军团是一个N行M列的矩阵，每个单元是一个血色先锋军的成员。感染瘟疫的人，每过一个小时，就会向四周扩散瘟疫，直到所有人全部感染上瘟疫。你已经掌握了感染源的位置，任务是算出血色先锋军的领主们感染瘟疫的时间，并且将它报告给巫妖王，以便对血色先锋军进行一轮有针对性的围剿。

输入输出格式

输入格式：

第1行：四个整数N，M，A，B，表示军团矩阵有N行M列。有A个感染源，B为血色敢死队中领主的数量。

接下来A行：每行有两个整数x，y，表示感染源在第x行第y列。

接下来B行：每行有两个整数x，y，表示领主的位置在第x行第y列。

【数据规模】

1<=M,N<=500

1<=A,B<=M*N

输出格式：

第1至B行：每行一个整数，表示这个领主感染瘟疫的时间，输出顺序与输入顺序一致。如果某个人的位置在感染源，那么他感染瘟疫的时间为0。

输入输出样例

输入样例#1：

5 4 2 3
1 1
5 4
3 3
5 3
2 4

输出样例#1：

3
1
3

说明

如下图，标记出了所有人感染瘟疫的时间以及感染源和领主的位置。

可是，由于没有发现领主们跟感染源站在一起（还有这么傻的？），对对对，我没有1A

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<iostream>
 4 #include<queue>
 5 using namespace std;
 6
 7 typedef pair<int,int> PII;
 8
 9 struct node{
10     int x,y,dep;
11     node(int x=0,int y=0,int dep=0):
12         x(x),y(y),dep(dep){}
13 };
14
15 const int maxn=505;
16
17 int n,m,n1,n2,get=0;
18 int site[maxn][maxn],vis[maxn][maxn];
19 int ans[maxn*maxn];
20 int dx[4]={0,-1,0,1},dy[4]={-1,0,1,0};
21 queue<node> que;
22
23 void init(){
24     scanf("%d%d%d%d",&n,&m,&n1,&n2);
25     for(int i=1,x,y;i<=n1;i++){
26         scanf("%d%d",&x,&y);
27         vis[x][y]=1;
28         que.push(node(x,y,0));
29     }
30     for(int i=1,x,y;i<=n2;i++){
31         scanf("%d%d",&x,&y);
32         if(vis[x][y])  get++;
33         site[x][y]=i;
34     }
35 }
36
37 void bfs(){
38     while(1){
39         node pos=que.front();  que.pop();
40         int posx=pos.x,posy=pos.y,posdep=pos.dep;
41         for(int i=0;i<4;i++){
42             int nxtx=posx+dx[i],nxty=posy+dy[i];
43             if(nxtx>=1&&nxtx<=n&&nxty>=1&&nxty<=m)
44                 if(!vis[nxtx][nxty]){
45                     vis[nxtx][nxty]=1;
46                     que.push(node(nxtx,nxty,posdep+1));
47                     if(site[nxtx][nxty]){
48                         ans[site[nxtx][nxty]]=posdep+1;
49                         if(++get==n2)  return;
50                     }
51                 }
52         }
53     }
54 }
55
56 int main(){
57     init();
58     bfs();
59     for(int i=1;i<=n2;i++)
60         printf("%d\n",ans[i]);
61     return 0;
62 }