这是一道bfs+状态压缩的题目,由于之前在学校教的都是dfs,对bfs第一次接触感触颇深。
对于这种迷宫类题目+寻找东西,bfs是很有用滴。
对于拥有的钥匙采用二进制表示简直不能更机制了。
1<<10,10把钥匙,利用| 运算符,来表示拥有钥匙,&运算符来检查是否有该门的钥匙。第几把就把usekey>>几。
#include<iostream>
#include<queue>
#include<cstring>
using namespace std;
const int N=20;
int m,n,p,k,s;
int door[N][N][N][N];
int key[N][N];
bool vis[N][N][1<<11];
struct node{
int x,y;
int step;
int havekey;
};
int dir[4][2]={{1,0},{0,1},{-1,0},{0,-1}};
void init(){
memset(door,-1,sizeof(door));
memset(key,0,sizeof(key));
memset(vis,0,sizeof(vis));
}
int bfs(){
queue <node>Q;
node st;
st.x=1;
st.y=1;
st.step=0;
st.havekey=key[1][1];
Q.push(st);
while(!Q.empty()){
node cur =Q.front();
Q.pop();
if(cur.x==n&&cur.y==m)return cur.step;
for(int i=0;i<4;i++){
int tx=cur.x+dir[i][0];
int ty=cur.y+dir[i][1];
if(tx>=1 && tx<=n && ty>=1 && ty<=m){
int usekey=door[cur.x][cur.y][tx][ty];
if(usekey==0) continue;
if(usekey==-1||(cur.havekey>>usekey)&1){
node nod;
nod.x=tx;
nod.y=ty;
nod.step=cur.step+1;
nod.havekey=cur.havekey|key[tx][ty];
if(vis[nod.x][nod.y][nod.havekey]==0){
vis[nod.x][nod.y][nod.havekey]=1;
Q.push(nod);
}
}
}
}
}
return -1;
}
int main()
{
std::ios::sync_with_stdio(false);
while(cin>>n>>m>>p){
init();
cin>>k;
int ax,bx,ay,by,g;
for(int i=0;i<k;i++){
cin>>ax>>ay>>bx>>by>>g;
door[ax][ay][bx][by]=door[bx][by][ax][ay] = g;
}
cin>>s;
int x,y,q;
for(int i=0;i<s;i++){
cin>>x>>y>>q;
key[x][y]|=(1<<q);
}
cout<<bfs()<<endl;
}
return 0;
}
借鉴了网上大神的代码还可以啦。。