poj 1204 Word Puzzles

//poj 1204 Word Puzzles
/*
题解：先把要询问的字符串用trie存起来，然后扫一遍字符矩阵，
      如果对应tire中且未回答则记录答案，最后把全部答案输出。
PS:   这题又让我学了不少，对于这种题可以像tarjan算法一样把
      输入的问题先存起来，再在搜索过程中慢慢回答。
*/
#include <iostream>
#include <vector>
using namespace std;
const int inf = 1<<28;
int n,m,k;

char str[1010][1010];
char a[1010];
int move[8][2]={ -1,0,-1,1,0,1,1,1,1,0,1,-1,0,-1,-1,-1};

struct Ans  //答案
{
       int x,y,d;
}ans[1010];
struct node  //字典树
{
       int branch[26];
}trie[1000010];
int mark[1000010],size;  //mark标记字典树有数据的结点,本题记录第几个字符串
void insert(char key[],int pos) //插入字典树
{
     int cur=0,j;
     for (int i=0;key[i];i++)
     {        
               j=key[i]-'A';
               if (trie[cur].branch[j]==-1)
               {
                          trie[cur].branch[j]=size;              
                          memset(trie[size].branch,-1,sizeof(trie[size].branch));
                          size++;
               }
               cur=trie[cur].branch[j];
     }
     mark[cur]=pos;
}
void check(int x,int y,int d)
{
     int i=x,j=y,cur=0;
     while(i>=0 && i<n && j>=0 && j<m && trie[cur].branch[str[i][j]-'A']!=-1)
     {
                cur=trie[cur].branch[str[i][j]-'A'];
                if (mark[cur] && ans[mark[cur]].d==-1)
                {
                              ans[mark[cur]].d=d;
                              ans[mark[cur]].x=x;
                              ans[mark[cur]].y=y;
                }    
                i+=move[d][0],j+=move[d][1];
     }
}
void solve()
{
     int v,w;
     for (int i=0;i<n;++i)
     for (int j=0;j<m;++j)
     for (int k=0;k<8;++k)
     {
         check(i,j,k);
     }
     for (int i=1;i<=k;i++) printf("%d %d %c/n",ans[i].x,ans[i].y,ans[i].d+'A');
}
int main()
{
    scanf("%d%d%d",&n,&m,&k);
    for (int i=0;i<n;i++) scanf("%s",str[i]);
    memset(ans,-1,sizeof(ans));
    memset(trie[0].branch,-1,sizeof(trie[0].branch));
    size=1;
    for (int j=1;j<=k;j++)
    {
          scanf("%s",a);
          insert(a,j);
    }   
    solve();
    system("pause");
    return 0;
   
}