UVA 10391 Compound Words

大意：判断一个字符串是否由给定的一些字符中两个组成。

思路：可以用hash，也可以用字典树+枚举，字典树似乎比较快。

CODE：

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cstdlib>
using namespace std;

#define MAXN 120010

char save[120010][30];

struct Trie
{
Trie *next[26];
int value;
Trie()
{
for(int i = 0; i < 26; i++)
next[i] = 0;
value = 0;
}
};

void insert(Trie *&root, char *s)
{
int i = 0, branch = 0;
Trie *p = root;
if(!p)
{
p = new Trie();
root = p;
}
while(s[i])
{
branch = s[i]-'a';
if(!p->next[branch]) p->next[branch] = new Trie();
p = p->next[branch];
i++;
}
p->value = 1;
}

int find(Trie *root, char *s)
{
int i = 0, branch = 0;
Trie *p = root;
if(!p) return 0;
while(s[i])
{
branch = s[i]-'a';
if(!p->next[branch]) return 0;
p = p->next[branch];
i++;
}
if(p->value) return 1;
return 0;
}

int main()
{
int tot = 0;
Trie *root = NULL;
while(~scanf("%s", save[tot]))
{
insert(root, save[tot]);
tot++;
}
for(int i = 0; i < tot; i++)
{
int len = strlen(save[i]);
for(int j = 0; j < len; j++)
{
char temp1[110] = {'\0'};
char temp2[110] = {'\0'};
strncpy(temp1, save[i], j); //前半部分
strncpy(temp2, save[i]+j, len-j); //后半部分
if(find(root, temp1) && find(root, temp2))
{
printf("%s\n", save[i]);
break;
}
}
}
return 0;
}