HDU 1274 Hat’s Words

http://acm.hdu.edu.cn/showproblem.php?pid=1247

字典树的题目，以前自己没想到怎么做就一直都放在那里了。

今天突然看到，然后想了下，自己有了思路然后就写了下。虽然WA了几次，但总算是让我过了。。。~~o(>_<)o ~~

#include <iostream>
#include <algorithm>
#include <string>
using namespace std;
const int maxn = 50005;
string str[maxn],s[maxn];
int num;

typedef struct node
{
    int f;
    node *next[26];
}*link;
link head;

int dtree(string str,int k,int n,int r)
{
    int i,j,f,flag;
    i=k;
    f=0;
    flag=1;
    link p,now;
    p=head;
    for(;i<n;i++)
    {
        if(!p->next[str[i]-'a'] && r)
        {
            now=(link)malloc(sizeof(node));
            now->f=0;
            for(j=0;j<26;j++)
                now->next[j]=NULL;
            now->f=0;
            p->next[str[i]-'a']=now;
            p=now;
        }
        else if(!p->next[str[i]-'a'] && !r)
        {
            return 0;
        }
        else
        {
            p=p->next[str[i]-'a'];
            if(p->f && r && !f)
            {
                f=dtree(str,i+1,n,!r);
            }
        }
    }
    if(!p->f && r)
    {
        p->f=1;
    }
    if(p->f && !r)
    {
        return 1;
    }
    else return f;  
}

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

bool cmp_2(string s1,string s2)
{
    return s1<s2;
}

void init()
{
    int i;
    head=(link)malloc(sizeof(node));
    head->f=0;
    for(i=0;i<26;i++)
        head->next[i]=NULL;
}

int main()
{
    int i,n,f,len;
    i=0;
    init();
    while(cin>>str[i])
    {
        i++;
        //if(str[i-1]=="sssss")
        //    break;
    }
    n=i;
    sort(str,str+n,cmp_1);
    num=0;
    for(i=0;i<n;i++)
    {
        len=str[i].length();
        f=dtree(str[i],0,len,1);
        if(f)
        {
            s[num++]=str[i];
        }
    }
    sort(s,s+num,cmp_2);
    for(i=0;i<num;i++)
        cout<<s[i]<<endl;
    return 0; 
}

1.先把单词按照长度排序，然后依次放入字典树。

2.在单词放入字典树的过程中，如果这个单词的第K个字母是某个单词的结尾，则将第[K...N-1]个字母当做一个单词string，放入字典树中查询。

3.如果要查询到string结尾，并且恰好是某个单词的结尾，则说明这也是一个单词，说明恰好是两个单词的组成；

   如果查询到string结尾，但是却并不是某个单词的结尾，则说明string只是某个单词的子串，并不是某个单词，所以返回string不存在。

4.如果在查询的过程中要创建新的节点，则说明string是一个新的单词，所以不用创建之间就返回string不存在。