问题来源
BYVoid魔兽世界模拟赛
【问题描述】
巫妖王的天灾军团终于卷土重来,血色十字军组织了一支先锋军前往诺森德大陆对抗天灾军团,以及一切沾有亡灵气息的生物。孤立于联盟和部落的血色先锋军很快就遭到了天灾军团的重重包围,现在他们将主力只好聚集了起来,以抵抗天灾军团的围剿。可怕的是,他们之中有人感染上了亡灵瘟疫,如果不设法阻止瘟疫的扩散,很快就会遭到灭顶之灾。大领主阿比迪斯已经开始调查瘟疫的源头。原来是血色先锋军的内部出现了叛徒,这个叛徒已经投靠了天灾军团,想要将整个血色先锋军全部转化为天灾军团!无需惊讶,你就是那个叛徒。在你的行踪败露之前,要尽快完成巫妖王交给你的任务。
军团是一个N行M列的矩阵,每个单元是一个血色先锋军的成员。感染瘟疫的人,每过一个小时,就会向四周扩散瘟疫,直到所有人全部感染上瘟疫。你已经掌握了感染源的位置,任务是算出血色先锋军的领主们感染瘟疫的时间,并且将它报告给巫妖王,以便对血色先锋军进行一轮有针对性的围剿。
【输入格式】
第1行:四个整数N,M,A,B,表示军团矩阵有N行M列。有A个感染源,B为血色敢死队中领主的数量。
接下来A行:每行有两个整数x,y,表示感染源在第x行第y列。
接下来B行:每行有两个整数x,y,表示领主的位置在第x行第y列。
【输出格式】
第1至B行:每行一个整数,表示这个领主感染瘟疫的时间,输出顺序与输入顺序一致。
如果某个人的位置在感染源,那么他感染瘟疫的时间为0。
【输入样例】
5 4 2 3
1 1
5 4
3 3
5 3
2 4
【输出样例】
3
1
3
【数据说明】
如下图,标记出了所有人感染瘟疫的时间以及感染源和领主的位置。
1 2 3 4 1 1 2 3 2 1 2 3 3 2 3 2 4 3 3 2 1 5 3 2 1<=M,N<=500 1<=A,B<=M*N
分析
一道简单的单源最短路模型转换。既然是单源最短路,源点只能有一个,可是题目中有多个感染源,怎么解决呢?我们可以虚拟一个点,这个点与各个源点之间连边,边权为0,这样只需要把这个虚拟点初始时加入队列,就可以求出整张图的最短路了。这就是这个题目中值得着重理解的转化思路。但是我在实际代码中没有虚拟这一个点,而是直接把感染源加入队列,并把他的dist设为0,这样就和前面所说的思路是等价的了。借助这个思路,我在考试时AC了这题,下面是代码:
/*
ID: ringxu97
LANG: C++
TASK: scarlet
SOLUTION: 最短路
*/
#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<cstdlib>
#include<algorithm>
#include<vector>
#include<stack>
#include<queue>
using namespace std; const int inf=0x3f3f3f3f; const int maxn=500+10;
const int maxa=maxn*maxn; const int dx[]={0,0,0,1,-1};
const int dy[]={0,1,-1,0,0}; struct POINT//定义图中节点
{
int x,y;
POINT(int a,int b){x=a;y=b;}
POINT(){x=0;y=0;}
}l[maxa],s[maxa];//l储存leader(领主) s储存Source(感染源)
int N,M,A,B;
int dist[maxn][maxn];//dist[i][j]表示传播到[i][j]所需最短时间
bool inq[maxn][maxn];
queue<POINT>Q; void init()
{
memset(inq,0,sizeof(inq));
for(int i=1;i<=N;++i)
for(int j=1;j<=M;++j)
{
dist[i][j]=inf;
}
} void read()//读入数据
{
scanf("%d%d%d%d",&N,&M,&A,&B);
init();
for(int i=1;i<=A;++i)
{
scanf("%d%d",&s[i].x,&s[i].y);
dist[s[i].x][s[i].y]=0;
POINT tmp(s[i].x,s[i].y);
inq[s[i].x][s[i].y]=1;//初始化源点,并将其加入队列
Q.push(tmp);
}
for(int i=1;i<=B;++i)
{
scanf("%d%d",&l[i].x,&l[i].y);
}
}
inline bool check(int i,int j)//检查是否在图内
{
return (1<=i && i<=N && 1<=j && j<=M);
}
void SPFA()//求最短路
{
while(!Q.empty())
{
POINT u=Q.front();Q.pop();
inq[u.x][u.y]=0;
for(int k=1;k<=4;++k)if(check(u.x+dx[k],u.y+dy[k]))
{
POINT v(u.x+dx[k],u.y+dy[k]);
if(dist[v.x][v.y]>dist[u.x][u.y]+1)
{
dist[v.x][v.y]=dist[u.x][u.y]+1;
if(!inq[v.x][v.y])
{
Q.push(v);
inq[v.x][v.y]=1;
}
}
} }
}
void print()//打印方案
{
for(int i=1;i<=B;++i)
{
printf("%d\n",dist[l[i].x][l[i].y]);
}
} int main()
{
freopen("scarlet.in", "r", stdin);
freopen("scarlet.out", "w", stdout);
read();
SPFA();
print();
return 0;
}