Problem
公主被恶人抓走,被关押在牢房的某个地方。牢房用N*M (N, M <= 200)的矩阵来表示。矩阵中的每项可以代表道路(@)、墙壁(#)、和守卫(x)。
英勇的骑士(r)决定孤身一人去拯救公主(a)。我们假设拯救成功的表示是“骑士到达了公主所在的位置”。由于在通往公主所在位置的道路中可能遇到守卫,骑士一旦遇到守卫,必须杀死守卫才能继续前进。
现假设骑士可以向上、下、左、右四个方向移动,每移动一个位置需要1个单位时间,杀死一个守卫需要花费额外的1个单位时间。同时假设骑士足够强壮,有能力杀死所有的守卫。
给定牢房矩阵,公主、骑士和守卫在矩阵中的位置,请你计算拯救行动成功需要花费最短时间。
Input
第一行为一个整数S,表示输入的数据的组数(多组输入)
随后有S组数据,每组数据按如下格式输入
1、两个整数代表N和M, (N, M <= 200).
2、随后N行,每行有M个字符。"@"代表道路,"a"代表公主,"r"代表骑士,"x"代表守卫, "#"代表墙壁。
Output
如果拯救行动成功,输出一个整数,表示行动的最短时间。
如果不可能成功,输出"Impossible"
Sample Input
2
7 8
#@#####@
#@a#@@r@
#@@#x@@@
@@#@@#@#
#@@@##@@
@#@@@@@@
@@@@@@@@
13 40
@x@@##x@#x@x#xxxx##@#x@x@@#x#@#x#@@x@#@x
xx###x@x#@@##xx@@@#@x@@#x@xxx@@#x@#x@@x@
#@x#@x#x#@@##@@x#@xx#xxx@@x##@@@#@x@@x@x
@##x@@@x#xx#@@#xxxx#@@x@x@#@x@@@x@#@#x@#
@#xxxxx##@@x##x@xxx@@#x@x####@@@x#x##@#@
#xxx#@#x##xxxx@@#xx@@@x@xxx#@#xxx@x#####
#x@xxxx#@x@@@@##@x#xx#xxx@#xx#@#####x#@x
xx##@#@x##x##x#@x#@a#xx@##@#@##xx@#@@x@x
x#x#@x@#x#@##@xrx@x#xxxx@##x##xx#@#x@xx@
#x@@#@###x##x@x#@@#@@x@x@@xx@@@@##@@x@@x
x#xx@x###@xxx#@#x#@@###@#@##@x#@x@#@@#@@
#@#x@x#x#x###@x@@xxx####x@x##@x####xx#@x
#x#@x#x######@@#x@#xxxx#xx@@@#xx#x#####@
Sample Output
13
7
这道题的话,我一开始用的是朴素版的bfs,当然很不幸wa了后面的数据。
仔细一想,关键问题其实是在于迷宫中守卫的设置。
“举一个例子,你正前方有三个守卫,你要走到正前方第四个格子,走直线的话,花费的时间就是7,假设旁边都没有守卫,你绕开走到的话,花费的时间仅为6。所以要使用优先队列。这样输出的就是最短时间了。”
而且我们要使用的是小根堆,这样每次从队头取出来的元素就是花费时间最小的那一个了,STL中有一个很好用的数据类型叫做pair二元组,它其实就相当于一个重载了小于号的结构体,排序时默认按照二元组中的第一个元素排序。
通过重载运算符可以定义时间短的优先级更高,最终找到正确结果。
在这里我们可以设置一个嵌套的pair型,即pair<int,pair<int,int>>
我的代码中简写成了pip
其中第一个类型int表示step步数,即时间,第二个类型表示搜索到的迷宫中的坐标。
当这个类型的变量压入优先队列中(我们应当使用小根堆:priority_queue<pip ,vector< pip >, greater< pip >>)时,就默认按照首元素step排序,堆顶即为step最小的元素。类似堆优化版的dijkstra的思想,每次利用堆顶进行扩展。
在bfs的过程中:如果遇到卫兵,走两步,如果是路,走一步,如果遇到公主,输出走过的最短步数step。
#include<bits/stdc++.h>
using namespace std;
const int N = 210;
char g[N][N];
int n,m;
typedef pair<int,int> pii;
typedef pair<int,pii> pip;
#define x first
#define y second
priority_queue<pip,vector<pip>,greater<pip>> pp;//小根堆
pip R;
bool st[N][N];
int s,sx,sy;
int dx[]={1,0,-1,0},dy[]={0,1,0,-1};
void bfs()
{
while(!pp.empty())
{
R = pp.top();
pp.pop();
for(int i=0;i<4;++i)
{
int step = R.x;
int a = R.y.x+dx[i],b = R.y.y+dy[i];
if(a<0||a>=n||b<0||b>=m) continue;
if(st[a][b]) continue;
if(g[a][b]=='#') continue;
if(g[a][b]=='a')
{
step+=1;
cout<<step<<endl;
return ;
}
if(g[a][b]=='x')
{
st[a][b]=true;
step+=2;
pp.push({step,{a,b}});
continue;
}
else
{
st[a][b]=true;
step+=1;
pp.push({step,{a,b}});
continue;
}
}
}
cout<<"Impossible"<<endl;
}
int main()
{
cin>>s;
while(s--)
{
memset(st,false,sizeof st);
while(!pp.empty()) pp.pop();//清空优先队列
cin>>n>>m;
for(int i=0;i<n;++i)
{
for(int j=0;j<m;++j)
{
cin>>g[i][j];
if(g[i][j]=='r')
{
int dist = 0;
sx = i,sy = j;
pp.push({dist,{sx,sy}});
st[sx][sy]=true;
}
}
}
bfs();
}
return 0;
}