poj 1204 Word Puzzles(字典树)

题目链接：http://poj.org/problem?id=1204

思路分析：由于题目数据较弱，使用暴力搜索；对于所有查找的单词建立一棵字典树，在图中的每个坐标，往8个方向搜索查找即可；

需要注意的是查找时不能匹配了一个单词就不在继续往该方向查找，因为在某个坐标的某个方向上可能会匹配多个单词，所以需要一直

查找直到查找到该方向上最后一个坐标；

代码如下：

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;

const int KIND = ;
const int MAX_N =  + ;
int row, column, num_word;
char map[MAX_N][MAX_N], words[MAX_N];
int dir[][] = {-, , -, , , , , , , , , -, , -, -, -};
int pos[MAX_N][];

struct Node {
    Node *next[KIND];
    bool end;
    int order;
    Node()
    {
        memset(next, , sizeof(next));
        end = false;
        order = ;
    }
};

void CreateTrie(Node *root, char *str, int j)
{
    int i = ;
    Node *p = root;

    while (str[i]) {
        int value = str[i] - 'A';
        if (p->next[value] == NULL)
            p->next[value] = new Node();
        p = p->next[value];
        ++i;
    }
    p->end = true;
    p->order = j;
}

void Search(Node *root, int i, int j, int k)
{
    Node *p = root;
    int pos_i = i, pos_j = j, value = ;

    while (p &&  <= pos_i && pos_i < row &&  <= pos_j && pos_j < column) {
        value = map[pos_i][pos_j] - 'A';
        if (p->next[value]) {
            if (p->next[value]->end) {
                pos[p->next[value]->order][] = i;
                pos[p->next[value]->order][] = j;
                pos[p->next[value]->order][] = k;
            }
            pos_i += dir[k][];
            pos_j += dir[k][];
            p = p->next[value];
        } else
            break;
    }
}

int main()
{
    while (scanf("%d %d %d\n", &row, &column, &num_word) != EOF) {
        Node *root = new Node;

        for (int i = ; i < row; ++i) {
            for (int j = ; j < column; ++j)
                scanf("%c", &map[i][j]);
            scanf("\n");
        }
        for (int i = ; i < num_word; ++i) {
            scanf("%s", words);
            CreateTrie(root, words, i);
        }
        for (int i = ; i < row; ++i)
            for (int j = ; j < column; ++j)
                for (int k = ; k < ; ++k)
                    Search(root, i, j, k);
        for (int i = ; i < num_word; ++i)
            printf("%d %d %c\n", pos[i][], pos[i][], pos[i][] + 'A');
    }
    return ;
}