HDU 1247 Hat’s Words（字典树）

 

题目大意

 

字典中，一个单词是一个 hat's word：这个单词可以由字典中另外两个单词拼接而成

给你一个字典，求出字典中所有的 hat's word

单词最多有 50000 个

 

做法分析

 

看到这道题，感觉无从下手，单词的数量这么多，但是仔细一想，单词数是多了，但是如果每个单词的平均长度必然就小了，不然光是读入数据就得超时

于是，先乱搞一发再说

 

建立一棵字典树，然后，对于每个单词，直接暴力，竟然神奇的过了！

暴力的做法是：枚举第一个子单词的长度，剩下的为第二个子单词，分别在字典树中匹配，看能否找到一个单词与两个单词匹配，能的话，这是一个 hat's word

 

参考代码

 

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

struct Tire
{
    struct Node
    {
        Node *next[26];
        bool flag;
        Node()
        {
            for(int i=0; i<26; i++) next[i]=NULL;
            flag=0;
        }
    };
    Node *root;
    Tire()
    {
        root=new Node();
    }

    void Insert(char s[])
    {
        Node *u=root;
        for(int i=0; s[i]; i++)
        {
            int v=s[i]-'a';
            if(u->next[v]==NULL) u->next[v]=new Node();
            u=u->next[v];
        }
        u->flag=1;
    }

    bool Find(char buff[], int s, int t)
    {
        Node *u=root;
        for(int i=s; i<=t; i++)
        {
            int v=buff[i]-'a';
            if(u->next[v]==NULL) return false;
            u=u->next[v];
        }
        return u->flag;
    }
} tree;

char s[50005][100];
int tot;

int main()
{
    for(tot=0; scanf("%s", s[tot])!=EOF; tot++)
        tree.Insert(s[tot]);
    for(int i=0; i<tot; i++)
    {
        int t=(int)strlen(s[i])-1;
        for(int j=0; j<t; j++)
            if(tree.Find(s[i], 0, j) && tree.Find(s[i], j+1, t))
            {
                printf("%s\n", s[i]);
                break;
            }
    }
    return 0;
}

AC通道

HDU 1247 Hat’s Words