lcp计数 （Trie）

lcp

思路：
看着挺唬人的，其实是道水题。
考虑每个子串作为lcp的贡献，就是包含这个子串的字符串的两两组合。

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <queue>
#include <iostream>
#define N 1000005
#define LL long long
using namespace std;

int n, rot, cnt; 
int c[N][26], sum[N];
char s[N];
LL ans;

void insert(char *s){
    int p = rot, len = strlen(s);
    for(int i=0; i<len; i++){
        int x = c[p][s[i] - 'a'];
        if( !x ) c[p][s[i] - 'a'] = ++cnt;
        p = c[p][s[i] - 'a'];
        sum[p]++;
    }
}

void query(){
    int p = rot;
    queue<int> q;
    q.push( p );
    while( !q.empty() ){
        int u = q.front(); q.pop();
        for(int i=0; i<26; i++){
            int v = c[u][i];
            if( !v ) continue;
            ans += (1LL * sum[v] * (sum[v] - 1) / 2);
            //一共有sum[v]个字符串中，包含v这个子串
            //那么v作为lcp的贡献就有C(2,sum[v]) 
            q.push( v );
        }
    }
}

int main(){
    scanf("%d", &n);
    rot = cnt = 1;
    for(int i=1; i<=n; i++){
        scanf("%s", s);
        insert( s );
    }
    query();
    cout << ans << endl;
}