码迷,mamicode.com
首页 > 其他好文 > 详细

poj-2001(字典树)

时间:2018-08-22 01:12:22      阅读:128      评论:0      收藏:0      [点我收藏+]

标签:def   pos   题意   trie   ring   poj   for   abc   pre   

题意:给你一堆字符串,我们定义一个字符串可以被缩写成一个字符串(必须是原字符串的前缀),问你每个字符串能辨识的前缀是什么,不能辨识意思是(ab,abc我们缩写成ab);

解题思路:可以用字典树解决,我们把刚开始的串存进去,然后在询问的时候,如果当前节点的字符只被一个串走过,那么肯定可以辨识;如果遍历完了找不到·,那么输出他自己;

代码:

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<string>
#include<cstring>
#define maxn 50050
using namespace std;
int trie[maxn][30];
string s[2050];
string a;
int cnt[maxn];
int root;
int tot;
void build_trie(string s)
{
    root=0;
    int slen=s.size();
    for(int i=0;i<slen;i++)
    {
        int id=s[i]-‘a‘;
        if(!trie[root][id])
        {
            trie[root][id]=++tot;
        }
        root=trie[root][id];
        cnt[root]++;
    }
}
int query(string s)
{
    root=0;
    int slen=s.size();
    for(int i=0;i<slen;i++)
    {
        int id=s[i]-‘a‘;
        if(cnt[root]==1)
        {
            return i-1;
        }
        root=trie[root][id];
    }
    return slen-1;
}
int main()
{
    int cot=0;
    while(cin>>a)
    {
        s[++cot]=a;
        build_trie(a);
    }
    for(int i=1;i<=cot;i++)
    {
        int pos=query(s[i]);
        cout<<s[i]<<" ";
        for(int j=0;j<=pos;j++)
            cout<<s[i][j];
        cout<<endl;
    }
}

  

poj-2001(字典树)

标签:def   pos   题意   trie   ring   poj   for   abc   pre   

原文地址:https://www.cnblogs.com/huangdao/p/9515315.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!