题目描述
自从LZY被大魔王抓走后,CYB成员们一直想尽方法找到他!
想要知道LZY的下落,只能破解出大魔王的密码。
现在给定一个二维网格和一个单词密码,找到该密码是否存在于网格中!
输入
题目有多组测试样例,第一行输入两个数字n 和 m,代表地图为n行m列,第二行需要查找的单词s,接下来有一个n*m的字符矩阵! (1 <= n, m <= 1000, 1 <= s.length() <= 100)
提示: 测试样例T (1 <= T <= 50)
输出
对于每个测试样例,在一行中输出是否能找到,如果是输出’True’, 否则输出’False’ (不包含单引号)
样例输入
3 4
ABCCED
ABCE
SFCS
ADEE
3 4
ABCB
ABCE
SFCS
ADEE
样例输出
True
False
总结:这题是一道搜索题,我发现搜索真的很好用,好多题目搜一搜就完事了,但是不是随便搜搜就好啦,还得考虑时间的问题,倒是我时间超限、内存超限好多次,真是难受死了,所以优化代码的时间效率是我接下来做题的目的。
代码 (DFS)
#include<bits/stdc++.h>
using namespace std;
string t;
char s[1005][1005];
int n,m,flag,a[1005],g,b[1005],dir[4][2]={1,0,-1,0,0,1,0,-1};
bool vis[1005][1005];
void dfs(int x,int y,int l){
if(flag==1) return;
if(l==t.size()){
flag=1;
return;
}
for(int i=0;i<4;i++){
int xx=x+dir[i][0];
int yy=y+dir[i][1];
if(xx>=0&&xx<n&&yy>=0&&yy<m&&!vis[xx][yy]&&s[xx][yy]==t[l]){
vis[xx][yy]=1;
dfs(xx,yy,l+1);
vis[xx][yy]=0;
}
}
return;
}
int main(){
ios::sync_with_stdio(false);
while(cin>>n>>m){
cin>>t;
flag=0;
g=0;
memset(a,0,sizeof(a));
memset(b,0,sizeof(b));
for(int i=0;i<n;i++){
for(int j=0;j<m;j++){
cin>>s[i][j];
if(s[i][j]==t[0]){
a[g]=i;
b[g]=j;
g++;
}
}
}
for(int i=0;i<g;i++){
memset(vis,0,sizeof(vis));
vis[a[i]][b[i]]=1;
dfs(a[i],b[i],1);
if(flag==1) break;
}
if(flag==1) cout<<"True"<<endl;
else cout<<"False"<<endl;
}
return 0;
}