UVAlive5792 Diccionário Portuñol

题目

题意:给出P个字符串,从所有字符串的前缀中任选一个,再给T个字符串,从所有字符串的后缀中任选一个,将其拼起来,能得到多少个不同的字符串。

题解:

考虑什么时候会发生重复出现字符串的情况:若P中有ab,T中有bc,那么a+bc和ab+c,产生了两个abc。

所以将T中所有字符串倒置,建立trie树,就可以知道有多少个不同后缀是以a开头,多少个不同后缀以b开头......,如上例中,以b开头的后缀有1个,以c开头的后缀有1个。

再将P中所有字符串建立trie树,考虑trie树的一条链aaab......,如果以aaa为当前P中选出的前缀,后面可以接以任意非b的字母开头的字符串,否则以aaab开头的字符串会被计算两次,依此方法遍历一遍trie树。要考虑的特殊情况,当trie树中某一条链为abcd,遍历到节点c时,若在T中字符串产生的trie树中,d与根节点相连,即存在后缀"d",不能将"abc"+"d"这种情况舍弃掉。

从上,不难发现,重复的次数=P中以字母i结尾的前缀的个数×T中以字母i开头的后缀的个数,'a'<=i<='z'

代码:

#include 
#include 
#include 
#include 
using namespace std;
#define maxn 100005

int p,s;
int trie[2][maxn][26],cnt1,cnt2;
char st[maxn];
int num[2][26];
void insert(int k,int &cnt)
{
    int node=0;
    for (int i=0;st[i];i++)
    {
        int tmp=st[i]-'a';
        if (!trie[k][node][tmp])
        {
            trie[k][node][tmp]=++cnt,num[k][tmp]++;
            for (int j=0;j<26;j++)
                trie[k][cnt][j]=0;
            if (!node) num[k][tmp]--;
        }
        node=trie[k][node][tmp];
    }
}
int main()
{
    //freopen("/home/christinass/code/in.txt","r",stdin);
    while (scanf("%d%d",&p,&s)!=-1)
    {
        if (!p&&!s) break;
        memset(num,0,sizeof(num));
        int root=0;
        cnt1=cnt2=0;
        for (int i=0;i<26;i++)
            trie[0][0][i]=trie[1][0][i]=0;
        for (int i=1;i<=p;i++)
        {
            scanf("%s",st);
            insert(0,cnt1);
        }
        for (int i=1;i<=s;i++)
        {
            scanf("%s",st);
            reverse(st,st+strlen(st));
            insert(1,cnt2);
        }
        long long ans=(long long)cnt1*cnt2;
        for (int i=0;i<26;i++)
            ans-=num[0][i]*num[1][i];
        printf("%lld\n",ans);
    }
    return 0;
}


你可能感兴趣的:(UVAlive5792 Diccionário Portuñol)