HDU 1247 Hat’s Words 经典字典树

来源:互联网 发布:java怎么写日志 编辑:程序博客网 时间:2024/05/22 23:27
题意
  给你若干个单词(不超过50000个),构成一个字典,输出这个字典中所有的帽子单词(Hat's word)。
  帽子单词(Hat's word):由两个已在字典中出现的单词构成。
思路
  根据输入的单词构建字典树,在插入的最后做一个这是“单词”的标记。我是用一个bool型来标记,默认为false,不是单词;如果为true,则说明到这个位置为止是一个单词。将所有单词记录下来,然后每一个单词将它拆分成两份,分别判断这两份是否为一个“单词”(在字典中出现过的字符串),这样每一个单词需要判断len-1次。例如,ahat,需要判断a和hat,ah和at,aha和t。第一组符合两份都是单词,所以这是一个帽子单词,输出该单词。
注意
  如果已经判断出这个单词是一个帽子单词,别忘了break退出循环,否则可能会重复输出。
#include<iostream>#include<string>#include<cstring>#include<cstdio>using namespace std;struct Tire{bool isw; //是否是一个单词Tire *next[26];Tire(){isw=false;memset(next,0,sizeof(next));}}root;string s[50005];int n=0;void insert(const string &s) //将word插入到字典树中,在最后标记这是一个单词{Tire *p=&root;int index;for(int i=0;s[i];i++){index=s[i]-'a';if(!p->next[index])p->next[index]=new Tire;p=p->next[index];}p->isw=true; //标记到这为止是一个单词}bool find(const string &s) //判断这个字符串是否为一个单词{Tire *p=&root;int index;for(int i=0;s[i];i++){index=s[i]-'a';if(!p->next[index]) return false;p=p->next[index];}return p->isw;}int main(){//freopen("E:\\ACM\\test.txt","r",stdin);while(cin>>s[n]){insert(s[n++]);}for(int i=1;i<n;i++){for(int j=1;j<s[i].size();j++){string s1=s[i].substr(0,j); //分割单词成两部分 string s2=s[i].substr(j,s[i].size());if(find(s1)&&find(s2)) {cout<<s[i]<<endl;break;}}}return 0;}


原创粉丝点击