题意:
给一个迷宫,求是否可从S走到G,迷宫中有门和钥匙,要获得相应种类的所有钥匙才能打开该种门(比如获得所有a才能打开A)。
分析:
枚举打开门的顺序,搜路径即可。
代码:
//poj 2157
//sep9
#include<iostream>
#include<algorithm>
using namespace std;
const int maxN=32;
char maze[maxN][maxN];
char g[maxN][maxN];
char order[6];
int key_num[256];
int n,m,start_i,start_j;
int vis[maxN][maxN];
int key_cnt;
void dfs(int i,int j,char ch)
{
if(vis[i][j]==1)
return ;
if(g[i][j]>='A'&&g[i][j]<='Z'&&g[i][j]!='S') return ;
vis[i][j]=1;
if(g[i][j]==ch) ++key_cnt;
if(j>0) dfs(i,j-1,ch);
if(j<m-1) dfs(i,j+1,ch);
if(i>0) dfs(i-1,j,ch);
if(i<n-1) dfs(i+1,j,ch);
}
bool find_key(char ch)
{
key_cnt=0;
memset(vis,0,sizeof(vis));
dfs(start_i,start_j,ch);
if(key_cnt==key_num[ch]) return true;
return false;
}
void delete_door(char ch)
{
for(int i=0;i<n;++i)
for(int j=0;j<m;++j)
if(g[i][j]==ch)
g[i][j]='.';
}
bool find_path(int i,int j)
{
if(vis[i][j]==1)
return false;
vis[i][j]=1;
if(g[i][j]=='G') return true;
if(g[i][j]>='A'&&g[i][j]<='Z'&&g[i][j]!='S') return false;
if(j>0&&find_path(i,j-1)) return true;
if(j<m-1&&find_path(i,j+1)) return true;
if(i>0&&find_path(i-1,j)) return true;
if(i<n-1&&find_path(i+1,j)) return true;
return false;
}
bool judge()
{
for(int i=0;i<n;++i)
for(int j=0;j<m;++j)
g[i][j]=maze[i][j];
for(int i=0;i<5;++i)
if(find_key(order[i]-'A'+'a'))
delete_door(order[i]);
memset(vis,0,sizeof(vis));
return find_path(start_i,start_j);
}
int main()
{
while(scanf("%d%d",&n,&m)==2){
if(m==0&&n==0) break;
for(int i=0;i<n;++i)
scanf("%s",maze[i]);
order[0]='A',order[1]='B',order[2]='C',order[3]='D',order[4]='E';
memset(key_num,0,sizeof(key_num));
for(int i=0;i<n;++i)
for(int j=0;j<m;++j){
++key_num[maze[i][j]];
if(maze[i][j]=='S')
start_i=i,start_j=j;
}
int ok=0;
do{
if(judge()){
ok=1;
break;
}
}while(next_permutation(order,order+5));
printf("%s\n",ok==1?"YES":"NO");
}
return 0;
}