题目链接:
http://acm.hdu.edu.cn/showproblem.php?pid=1247
题目大意:
给出一个单词表,询问其中是否存在某些单词,如hat-word,是由表中其他两个单词拼接得到的,输出所有的这样的单词
分析:
直接将单词全部插入字典树,然后暴力枚举每个单词的所有分割可能即可
代码:
#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cmath>
#include<map>
#include <string>
using namespace std;
typedef long long ll;
struct Trie{
Trie *next[26];
int val;
Trie(){
val = -1;
for (int i = 0 ; i < 26 ; i ++)
next[i] = NULL;
}
};
void addword(string& str,Trie* node,int i)
{
if (node->next[str[i]-'a']==NULL)
node->next[str[i]-'a'] = new Trie;
node = node->next[str[i]-'a'];
++i;
if (i<str.size())
addword(str,node,i);
else
{
node->val = 1;
return;
}
}
int query(string & str ,Trie *node,int i)
{
if (node->next[str[i]-'a']==NULL)
return -1;
else
node= node->next[str[i]-'a'];
++i;
if (i<str.size())
return query(str,node,i);
else
return node->val;
}
string str[100000];
int main()
{
int cnt=0;
string p,q;
Trie* head = new Trie;
ios::sync_with_stdio(false);
while (cin>>str[cnt])
{
addword(str[cnt],head,0);
cnt++;
}
sort(str,str+cnt);
for (int i = 0 ; i < cnt ; i ++)
{
p.clear();
q.clear();
for (int j = 0 ; j < str[i].size()-1 ;j ++)
{
p +=str[i][j];
q = str[i].substr(j+1);
//cout<<" p = "<<p<<" q = "<<q<<endl;
if (query(p,head,0)!=-1&&query(q,head,0)!=-1)
{
cout<<str[i]<<endl;
break;
}
}
}
return 0;
}