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;
}