Hdu 1247 - Hat’s Words （字典树）

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

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

0 ·

CC 4.0 BY-SA版权

字符串专栏收录该内容

5 篇文章

订阅专栏

本文介绍了一种利用字典树（Trie）的数据结构来高效解决单词拼接问题的方法。通过构建字典树并遍历单词列表，可以找出哪些单词能够由列表中的其他两个单词拼接而成。代码实现简洁明了，易于理解。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接：
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;
}