UVAlive - 5792 Diccionário Portuñol (Trie)

转自:cxlove
题意:有两个字符串集合,从第一个集合中取某个串的非空前缀,从第二个集合中取某个串的非空后缀,拼接成一个串,问有多少个不同的新串。
思路:首先把第一个集合中的串加入到Trie中,然后就可以枚举前缀了,然后就是统计后缀,麻烦的是可能出现重复的。

当我们按长度枚举前缀的时候,可以考虑如果后缀的前缀出现在前缀的后继中,那便不用考虑,后面总会统计到。

这样的话,其实只要比较第一个字母。

如果当前结点P表示一个前缀串,那么如果p -> next[i]为空,说明前缀中不含字母i为后继,那么我们就可以统计以i开头的后缀了。否则总能在p -> next[i]为前缀中统计到。

有一点需要注意的是,如果 p -> next[i]便是我们最终的串,即以p 为前缀时,后缀只有一个字母的话,那就不能把这一个字母拉入到前缀中,因为后缀是非空的。所以再处理下end[i],表示是否存在以单个字母i为后缀的。

至于统计以i开头的不重复后缀的话,可以反向加入到Trie中统计。

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<vector>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<algorithm>
using namespace std;
typedef long long LL;
const int maxn=100010;
const int maxm=1010;
const int SIGMA_SIZE=26;
int P,S;
char a[maxm][maxm],b[maxm][maxm];
int tail[maxn];
int cnt[30];
LL ans;
struct Trie
{
    int ch[maxn][SIGMA_SIZE];
    int val[maxn];
    int sz;
    void clear(){memset(ch[0],0,sizeof(ch[0]));sz=1;}
    int idx(char x){return x-'a';}
    void insert(char *s)
    {
        int u=0;
        for(int i=0;s[i];i++)
        {
            int c=idx(s[i]);
            if(!ch[u][c])
            {
                memset(ch[sz],0,sizeof(ch[sz]));
                val[sz]=0;
                ch[u][c]=sz++;
            }
            u=ch[u][c];
        }
        val[u]=1;
    }
    void cal(int u)
    {
        for(int i=0;i<SIGMA_SIZE;i++)
        {
            if(ch[u][i])
            {
                cnt[i]++;
                cal(ch[u][i]);
            }
        }
    }
    void solve(int u)
    {
        for(int i=0;i<SIGMA_SIZE;i++)
        {
            if(!ch[u][i])
            {
                if(u)
                ans+=cnt[i];
            }
            else
            {
                if(u&&tail[i])ans++;
                solve(ch[u][i]);
            }
        }
    }
}prefix,suffix;
int main()
{
    while(scanf("%d%d",&P,&S)!=EOF,P||S)
    {
        prefix.clear();
        suffix.clear();
        memset(cnt,0,sizeof(cnt));
        memset(tail,0,sizeof(tail));
        for(int i=0;i<P;i++)
        {
            scanf("%s",a[i]);
            prefix.insert(a[i]);
        }
        for(int i=0;i<S;i++)
        {
            scanf("%s",b[i]);
            reverse(b[i],b[i]+strlen(b[i]));
            tail[b[i][0]-'a']=1;
            suffix.insert(b[i]);
        }
        ans=0;
        suffix.cal(0);
        prefix.solve(0);
        cout<<ans<<endl;
    }
    return 0;
}
阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/u010660276/article/details/46814849
上一篇hdu 4760 - Good Firewall(Trie)
下一篇2014上海全国邀请赛(hdu 5090 - 5099)dp+线段树+拓扑排序+bfs(状态压缩)
想对作者说点什么? 我来说一句

RIO karma 传歌软件

2013年11月07日 19.72MB 下载

没有更多推荐了,返回首页

关闭
关闭