codeforces 596E Wilbur and Strings
题意:一个矩阵上面有0~9的数字,可以从任意一个格子出发,每次根据格子上的数字会前进到另一个格子(或原地不动),现在给出q个数位串,问是否有走法可以取出这个串(走到格子上的时候可以不取)。
思路:发现每个点的出度至多为1,那么可以预处理出走到这个格子后取某个数字最近的格子是哪个(f[i][j][0-9]),然后用队列存下开头,在做一遍贪心就可以了
#include<cstdio> #include<cmath> #include<cstring> #include<iostream> #include<algorithm> int cnt,sz=0,f[201*201][10],c[201][201],id[201][201],vis[201][201],n,m,a[10],b[10],ans[201*201],d[201*201]; int read(){ char ch=getchar();int f=1,t=0; while (ch<'0'||ch>'9') {if (ch=='-') f=-1;ch=getchar();} while ('0'<=ch&&ch<='9'){t=t*10+ch-'0';ch=getchar();} return t*f; } void dfs(int x,int y){ int dx=x+a[c[x][y]],dy=y+b[c[x][y]]; if (dx<1||dx>n||dy<1||dy>m) f[id[x][y]][c[x][y]]=id[x][y]; else{ f[id[x][y]][c[x][y]]=id[dx][dy]; if (!vis[dx][dy]) {vis[dx][dy]=1;dfs(dx,dy);} for (int i=0;i<=9;i++) if (i!=c[x][y]) f[id[x][y]][i]=f[id[dx][dy]][i]; } } int main(){ n=read();m=read();int T=read();cnt=0; for (int i=1;i<=n;i++){ char s[2005]; scanf("%s",s+1); for (int j=1;j<=m;j++){ c[i][j]=s[j]-'0';id[i][j]=++cnt;d[cnt]=c[i][j];
} }
char s[1000005]; for (int i=0;i<=9;i++) a[i]=read(),b[i]=read(); for (int i=1;i<=n;i++) for (int j=1;j<=m;j++) if (!vis[i][j]){ vis[i][j]=1; ans[++sz]=id[i][j]; dfs(i,j); }while (T--){ scanf("%s",s+1);bool flag=false; for (int i=1;i<=sz;i++){ int x=ans[i]; for (int j=1;j<=strlen(s+1);j++){ x=f[x][s[j]-'0'];if (x<=0) break; } if (x>0) {flag=true;break;} } if (flag) printf("YES\n");else printf("NO\n"); } }