HDOJ-1247Hat’s Words（Trie）

最新推荐文章于 2019-08-09 20:14:40 发布

原创最新推荐文章于 2019-08-09 20:14:40 发布 · 335 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#acm #hdoj #Trie

ACM 专栏收录该内容

76 篇文章

订阅专栏

本文介绍了一种使用Trie树的数据结构来检测给定单词是否能由其他两个单词组合而成的方法。通过将所有单词插入Trie树并检查每个单词是否可以被分割成两个子串，这两个子串分别存在于Trie树中。

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
#include <iostream>
#include <algorithm>
#define MAXN 26

using namespace std;

typedef struct Trie
{
    int next[MAXN];
    int cnt;
    bool tail;
};

Trie    f[250000];
string  s[50005];
int  tot;

bool cmp(const string s1,const string s2)
{
    return s1.length()<s2.length();
}

void TrieInit()
{
    int i;
    tot=MAXN;
    memset(f,0,sizeof(f));
    for(i=0;i<MAXN;i++) f[0].next[i]=i+1;
}


void Insert(string s)
{
    int i,j,k,l,flag;
    l=s.length();
    k=0;
    for(i=0;i<l;i++)
    {

        if(f[k].next[s[i]-'a']!=0)
        {
            k=f[k].next[s[i]-'a'];
            f[k].cnt++;
            if(i==l-1) f[k].tail=1;
        }
        else
        {
            tot++;
            f[k].next[s[i]-'a']=tot;
            k=tot;
            f[k].cnt++;
            if(i==l-1) f[k].tail=1;
        }
    }

}


bool Find(string s)
{
    int i,j,k,l,flag;
    l=s.length();
    k=0;    flag=0;
    for(i=0;i<l;i++)
    {

        if(f[k].next[s[i]-'a']!=0)
        {
            k=f[k].next[s[i]-'a'];
            if(i==l-1)
            {
                if(f[k].tail) return 1;
                    else return 0;
            }
        }
        else
        {
            return 0;
        }
    }
    return 0;
}




int main()
{
    int i,j,k,t,m,n;
    string temp,temp2;

    TrieInit();
    n=0;
    while(cin>>s[n])
    {
        Insert(s[n]);
        n++;
    }

    for(i=0;i<n;i++)
    {
        temp="";
        for(j=0;j<s[i].length()-1;j++)
        {
            temp+=s[i][j];
            temp2="";
            for(k=j+1;k<s[i].length();k++) temp2+=s[i][k];
            if(Find(temp) && Find(temp2))
            {
                cout<<s[i]<<endl;
                break;
            }
        }
    }


    return 0;
}

题意：

按照字典序给定n个单词，问那些单词是有其它两个单词组成的。

思路：

先把所有单词加入Trie树中，然后对每个单词进行拆分，判断拆出来的两部分能否在Trie中查询到。

代码：