字典树PKU 1204 Word Puzzles

题目的大意为：

要求编程实现查找字谜拼图中的单词，输出查找到的单词的起始位置和方向(ABCDEFGH分别代表北、东北、东、东南、南、西南、西、西北八个方向)。
输入:
输入的第一行包含三个正数：行数，0＜L＜=1000；列数，0＜C＜= 1000；和字谜单词个数0＜W＜＝1000。接下来的L行，每行输入C个字符。最后W行每行输入一个字谜单词。
输出:
你的程序应该每行输出每个单词（跟输入的顺序相同）第一个字母出现的行和列的下标。在三个值之间必须是一个空间分隔。

解题思路：

此题涉及到单词字符串的查找匹配，为了提高查找效率，减少空间耗费，可以考虑使用字典树，字典树的优点是能利用字符串的公共前缀来减少查询时间，最大限度地减少无谓的字符串比较。
字典树：
字典树的根节点不包含字符，除根节点外每一个节点都只包含一个字符； 从根节点到某一节点，路径上经过的字符连接起来，为该节点对应的字符串； 每个节点的所有子节点包含的字符都不相同。
针对该题，可以设计一个26个大写字母的字典树，结点结构为：

struct Node{
     int index;//字符串结尾标志，单词的序号,初始化为-1
     Node* next[26];//指向26个子树的指针数组,初始化为null
};

此处用到的操作有插入，查找;
字典树插入字符串的方法为：
(1)从根结点的子结点开始插入的字符串的第一字符,（若该子结点为空，创建该结点）,并将该结点作为下一个字符的父结点。
(2)迭代到下一个字符，直至字符串全部插入。
(3)在字符串最后一个字母的结点标记该字符串的序号，方便查找。
搜索字典项目的方法为：
(1) 从根结点开始一次搜索；
(2) 取得要查找关键词的第一个字母，并根据该字母选择对应的子树并转到该子树继续进行检索；
(3) 在相应的子树上，取得要查找关键词的第二个字母,并进一步选择对应的子树进行检索。
(4) 迭代过程……
(5) 在某个结点处，关键词的所有字母已被取出（index非负），即完成查找，若结点为空，查找失败。
例如：往一棵空树中插入序列:YES，假设该字符串的序号index为2

代码：

#include<stdio.h>
#include <memory.h>
#define MaxNode 26//子结点数量
char puzzle[1001][1001];
//A...H的8个方向
int D[8][2] = { { -1, 0 }, { -1, 1 }, { 0, 1 }, { 1, 1 },
{ 1, 0 }, { 1, -1 }, { 0, -1 }, { -1, -1 } };
struct {
    int x;
    int y;
    char dir;
}outputList[1001];//记录单词位置和方向的数组
struct TrieNode{//结点结构
public:
    int wordIndex;//单词序号
    TrieNode* next[MaxNode];//26个字母的子结点指针
    TrieNode(){//构造函数
        wordIndex = -1;
        memset(next, NULL, sizeof(TrieNode*)*MaxNode);
    }
};
class Trie{//字典树
public:
    int MaxRow, MaxCol, MaxWords;
    Trie()
    {
        root = new TrieNode();
    };
    int find(int i, int j,char dir)//从i行，j列沿dir方向查找
    {
        int pos = puzzle[i][j] - 'A';
        int x = i,y = j;
        TrieNode* current = root->next[pos];
            while (current != NULL)
            {
                if (current->wordIndex > 0)//找到单词，记录单词位置和方向
                {
                    outputList[current->wordIndex].x = x;
                    outputList[current->wordIndex].y = y;
                    outputList[current->wordIndex].dir = (char)dir + 'A';;
                }
                i += D[dir][0]; j += D[dir][1];//dir方向的下一个字母的下标
                if (i < 0||j<0||i>=MaxRow||j>=MaxCol)break;//超出puzzle图则跳出循环
                pos = puzzle[i][j] - 'A';
                current = current->next[pos];//指向下一个字典树结点
            }
        return -1;
    };
    void insert(char *des,int index)//插入序号为index的字符串des
    {
        TrieNode* current = root;//从根结点开始
        int pos;
        for (; *des!='\0'; des++)//将每个字符插入
        {
            pos = *des - 'A';
            if (current->next[pos] == NULL)
                current->next[pos] = new TrieNode();//结点为空则创建
            current = current->next[pos];
        }
        current->wordIndex = index;//插入结束，标记该字符串序号
    };
private:
    TrieNode * root;//树根指针
};
int main()
{
    Trie T;
    scanf("%d%d%d", &T.MaxRow, &T.MaxCol, &T.MaxWords);
    getchar();//去掉换行符
        for (int i = 0; i < T.MaxRow; i++)
        {
            gets(puzzle[i]);
        }
        char t[1001];
        for (int i = 1; i <= T.MaxWords; i++)
        {
            gets(t);
            T.insert(t,i);
        }
        for (int i = 0; i < T.MaxRow; i++)
        for (int j = 0; j < T.MaxCol; j++)
        {
            for (int dir = 0; dir <= 7; dir++)
            {
                T.find(i, j, dir);
            }
        }
        for (int i = 1; i <= T.MaxWords; i++)
        {
            printf("%d %d %c\n", outputList[i].x, outputList[i].y, outputList[i].dir);
        }
    return 0;
}