HDU 1247 Hat’s Words 字典树

最新推荐文章于 2021-08-05 13:35:01 发布

原创最新推荐文章于 2021-08-05 13:35:01 发布 · 556 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#HDU #1247 #Hats Words #字典树

ACM 专栏收录该内容

166 篇文章

订阅专栏

本文介绍了如何在字典中找到由两个单词组合而成的特殊单词（hat's words），通过建立字典树并进行字符串分割搜索来实现。关键点包括标记单词结束节点和避免重复输出。

Hat’s Words

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 11324 Accepted Submission(s): 4044

Problem Description

A hat’s word is a word in the dictionary that is the concatenation of exactly two other words in the dictionary.
You are to find all the hat’s words in a dictionary.

Input

Standard input consists of a number of lowercase words, one per line, in alphabetical order. There will be no more than 50,000 words.
Only one case.

Output

Your output should contain all the hat’s words, one per line, in alphabetical order.

Sample Input

a
ahat
hat
hatword
hziee
word

Sample Output

ahat
hatword

题意：寻找由其它单词组合得到的单词

思路：对所有的单词建立字典树，然后，对每一个单词暴力分割，将分割得到的两部分放入字典树中搜索，根据搜索结果判断。

坑点:1.建立字典树的时候，需要标记每一个单词的结束节点

2.如果搜索到一个单词符合条件，需要跳出循环，防止多次输出

#include <iostream>
#include <stdio.h>
#include <string.h>
#define MAX 26
using namespace std;

struct trie
{
    bool is_end;//标记该节点所表示的是不是完整的单词
    trie* next[MAX];
};
trie* root=new trie;
char words[50000][100];

void insert_node(char* str)
{
    trie *p,*newnode;
    p=root;
    for(int i=0;str[i]!='\0';i++)
    {
        if(p->next[str[i]-'a']==NULL)
        {
            newnode=new trie;
            newnode->is_end=false;
            for(int j=0;j<MAX;j++)
                newnode->next[j]=NULL;
            p->next[str[i]-'a']=newnode;
            p=newnode;
        }
        else
        {
            p=p->next[str[i]-'a'];
        }
    }
    p->is_end=true;
}


int find_str(char *str)
{
    trie *p=root;
    for(int i=0;str[i]!='\0';i++)
    {
        if(p->next[str[i]-'a']!=NULL)
            p=p->next[str[i]-'a'];
        else
            return 0;
    }
    if(p->is_end==true)
        return 1;
    else
        return 0;
}

void get_sub(char *dst,char *src, int s,int l)
{
    int i;
    for(i=s;i<s+l;i++)
        dst[i-s]=src[i];
    dst[i-s]='\0';
}


int main()
{

    char temp[20];
    int flag1,flag2,len;
    int p=0;
    for(int i=0;i<MAX;i++)
        root->next[i]=NULL;
    while(gets(words[p]))
    {
        if(words[0]=='\0')
            break;
        insert_node(words[p]);
        p++;
    }
    for(int i=0;i<p;i++)
    {
        len=strlen(words[i]);
        for(int j=1;j<len;j++)
        {
            get_sub(temp,words[i],0,j);
            flag1=find_str(temp);
            get_sub(temp,words[i],j,len-j);
            flag2=find_str(temp);
            if(flag1&&flag2)
            {
                printf("%s\n",words[i]);
                break;//注意需要跳出，防止重复输出
            }

        }
    }
    return 0;
}