hdu 1247(字典树)

/*
*   字典树
*   思路：把所有单词建成一棵字典树，然后把每个分成两部分，判断这两部分是否在树中
*         是，就符合条件。
*/

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>

using namespace std;

const int N = 16;
const int M = 50005;

char strL[N], strR[N], dic[M][N]; 
struct node {
    bool flag;
    node *child[26];
    node() {
        flag = false;
        for (int i=0; i<26; ++i) child[i] = NULL;
    }
}*root;

void insert(char str[]) {
    node *p = root;
    for (int k,i=0; str[i]; ++i, p=p->child[k]) {
        k = str[i] - 'a';
        if (!p->child[k]) p->child[k] = new node(); 
    }
    p->flag = true; //单词结尾 
}

bool search(char str[]) {
    node *p = root;
    for (int k,i=0; str[i]; ++i) {
        k = str[i] - 'a';
        if (p->child[k]) p = p->child[k];
        else return false;
    }
    return p->flag; //注意，可能表示单词的结尾，也可能不是 
}

int main() {
    int i, k, ni, nj, len;
    root = new node();
    for (i=-1; scanf("%s", dic[++i])!=EOF; insert(dic[i]));
    for (int j=0; j<=i; ++j) {
        len = strlen(dic[j]);
        for (k=1; k<len; ++k) {//把单词分成两部分 
            for (ni=0; ni<k; ++ni) strL[ni] = dic[j][ni];
            for (nj=k; nj<len; ++nj) strR[nj-k] = dic[j][nj];
            strL[ni] = strR[nj-k] = '\0';
            if (search(strL) && search(strR)) {
                printf ("%s\n", dic[j]);
                break;
            } 
        }
    }
    return 0;
}