【洛谷P1101】单词方阵
题目大意:给一 \(n \times n\) 的字母方阵,内可能蕴含多个 \(“yizhong”\) 单词。单词在方阵中是沿着同一方向连续摆放的。摆放可沿着 8 个方向的任一方向,同一单词摆放时不再改变方向,单词与单词之间可以交叉,因此有可能共用字母。输出时,将不是单词的字母用*代替,以突出显示单词。
题解:首先每个单词摆放的方向要相同。因此,搜素对应的状态应为当前位置的坐标,需要比对的第几个字母和面朝哪个方向。搜索时因为一个字母可能被多个字符串公用,因此,不是很方便确定一个字母是否会不被打印,但是若经过这个字母可以组成单词,则表明该字母一定在输出时会被打印出来。所以,可以采用一个 vis[][] 数组表示该字母是否需要被打印即可。
代码如下
#include <bits/stdc++.h>
using namespace std;
const int maxn=110;
char mp[maxn][maxn],s[maxn]="yizhong";
int n;bool vis[maxn][maxn];
void read_and_parse(){
scanf("%d",&n);
for(int i=1;i<=n;i++)scanf("%s",mp[i]+1);
}
const int dx[]={0,0,1,1,1,-1,-1,-1};
const int dy[]={1,-1,0,1,-1,0,1,-1};
bool dfs(int x,int y,int dir,int idx){
if(idx==6)return vis[x][y]=1;
int xx=dx[dir]+x,yy=dy[dir]+y;
if(xx>=1&&xx<=n&&yy>=1&&yy<=n&&mp[xx][yy]==s[idx+1]&&dfs(xx,yy,dir,idx+1))return vis[xx][yy]=1;
else return 0;
}
void solve(){
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
if(mp[i][j]=='y'){
for(int k=0;k<8;k++)
if(dfs(i,j,k,0))vis[i][j]=1;
}
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++)printf("%c",vis[i][j]?mp[i][j]:'*');
puts("");
}
}
int main(){
read_and_parse();
solve();
return 0;
}