（字典树）HDU

原题链接：http://acm.hdu.edu.cn/showproblem.php?pid=1247

---

题意：输出在所给字典中所有的hat's word。hat's word：是字典中由其他两个单词连接而成的的单词。

---

分析：现存下所有单词，然后遍历查询。用exist记录当前节点是否组成单词。当遍历某一个单词word时，如果在遍历结束前，在第i个查到当前前缀已经组成了单词，那么再从头遍历检查word.substr(i+1,word.length())【从第1个开到末尾组成的子串】。因为可能存在多个i，所以用res记录可以组合的方法数，最后判断如果res>0，则输出那个word。

因为两个单词检查判断不太一样，可以分两个函数写。

---

代码：

#include <set>
#include <map>
#include <list>
#include <cmath>
#include <queue>
#include <vector>
#include <bitset>
#include <string>
#include <cctype>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>

using namespace std;

typedef long long ll;
typedef unsigned long long ull;
#define inf (0x3f3f3f3f)
#define lnf (0x3f3f3f3f3f3f3f3f)
#define eps (1e-8)
int sgn(double a){return a < -eps ? -1 : a < eps ? 0 : 1;}

//--------------------------

struct  Trie_node {
    int cnt;
    int next[26];
    bool exist;
}tree[400010];
int nxt;

int  createTrieNode() {
    memset(&tree[nxt], 0, sizeof(Trie_node));
    return nxt++;
}

void Trie_insert(string &word) {
    int rt=0;
    int len=word.length();
    for(int i=0;i<len;i++){
        int id = word[i]-'a';
        if(!tree[rt].next[id]){
            tree[rt].next[id]=createTrieNode();
        }
        rt=tree[rt].next[id];
        tree[rt].cnt++;
    }
    tree[rt].exist=true;
}

bool Trie_search2(string word){
    int rt=0;
    int len=word.length();
    for(int i=0;i<len;i++){
        int id=word[i]-'a';
        if(!tree[rt].next[id])return false;
        rt=tree[rt].next[id];
    }
    return tree[rt].exist;
}

bool Trie_search1(string &word) {
    int rt=0;
    int len=word.length();
    int res=0;
    for(int i=0;i<len;i++){
        int id=word[i]-'a';
        rt=tree[rt].next[id];
        if(tree[rt].exist&&Trie_search2(word.substr(i+1,len-i-1)))res++;
    }
    return res>0;
}

void init(){
    nxt=1;
    memset(&tree[0],0,sizeof(Trie_node));
}

string word[50010];

void solve() {
    init();
    int n=0;
    while(cin>>word[n]){
        Trie_insert(word[n]);
        n++;
    }
    for(int i=0;i<n;i++){
        if(Trie_search1(word[i]))cout<<word[i]<<endl;
    }

}




int main() {

#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
    //freopen("out.txt", "w", stdout);
#endif
    //iostream::sync_with_stdio(false);
    solve();
    return 0;
}