字符串专项：LA 5913

一道tire树的应用，分别用前缀和后缀建tire树，tire树的边数（节点数-1）即为前后缀的总数，两者的乘积即为所有组合的总数。但是这样存在重复的组合，需要扣掉不与根相连的相同字母的边数乘积。最后还应考虑单个字母的情况。具体为什么这样，讲也很难讲清，仔细想一下应该可以明白。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
typedef long long LL;
#define M 400010
#define sigma 26
LL v1[sigma],v2[sigma];
int ff[sigma];
int ch1[M][sigma],ch2[M][sigma];
LL sz1,sz2;
void init()
{
    sz1=sz2=1;
    memset(ch1[0],0,sizeof(ch1[0]));
    memset(ch2[0],0,sizeof(ch2[0]));
    memset(ff,0,sizeof(ff));
    memset(v1,0,sizeof(v1));
    memset(v2,0,sizeof(v2));
}
int idx(char c){return c-'a';}
char str[50];
void insert(char *s)
{
    int u1=0,u2=0,n=strlen(s);
    if(n==1) ff[s[0]-'a']=1;
    for(int i=0;i<n;i++)
    {
        int c1=idx(s[i]),c2=idx(s[n-i-1]);
        if(!ch1[u1][c1])
        {
            memset(ch1[sz1],0,sizeof(ch1[sz1]));
            ch1[u1][c1]=sz1++;
            if(i) v1[c1]++;
        }
        u1=ch1[u1][c1];
        if(!ch2[u2][c2])
        {
            memset(ch2[sz2],0,sizeof(ch2[sz2]));
            ch2[u2][c2]=sz2++;
            if(i) v2[c2]++;
        }
        u2=ch2[u2][c2];
    }
}
int n;
int main()
{
    while(cin>>n)
    {
        init();
        for(int i=0;i<n;i++)
        {
            cin>>str;
            insert(str);
        }
        LL ans=(sz1-1)*(sz2-1);
        //cout<<sz1<<" "<<sz2<<endl;
        for(int i=0;i<sigma;i++)
        {
            ans-=v1[i]*v2[i];
            ans+=ff[i];
        }
        cout<<ans<<endl;
    }
    return 0;
}