UVa 1519 Dictionary Size

方法：Trie

看了题解，有两种做法，大致是相通的。这道题重点在于如何判重。

建立两个trie，取名prefix 和 suffix。把所有string插入第一个trie，每个节点就代表一种prefix。同理，把所有string反转之后插入第二个trie，每个节点就代表一个suffix。如果没有重复的话，那么结果就是prefix.size * suffix.size。如何判重呢？如果一个长度至少为2的前缀p以字符c结尾，并且有一个长度至少为2的后缀 s以c开始，那么 p+s.substring(1) == p.substring(0, p.length()-1) + s。 所以，将trie中每个节点所代表的字符的数量记录下来，记在cnt[26], 所用重复的个数即 sum(prefix.cnt[i] * suffix.cnt[i]) for i = 0,1,2,..,25。

注意，我们有漏掉一种dictionary word 长度为1的情况，所以要把这些加起来。

code:

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <string>
#include <vector>
#include <stack>
#include <bitset>
#include <cstdlib>
#include <cmath>
#include <set>
#include <list>
#include <deque>
#include <map>
#include <queue>
#include <fstream>
#include <cassert>
#include <unordered_map>
#include <unordered_set>
#include <cmath>
#include <sstream>
#include <time.h>
#include <complex>
#include <iomanip>
#define Max(a,b) ((a)>(b)?(a):(b))
#define Min(a,b) ((a)<(b)?(a):(b))
#define FOR(a,b,c) for (ll (a)=(b);(a)<(c);++(a))
#define FORN(a,b,c) for (ll (a)=(b);(a)<=(c);++(a))
#define DFOR(a,b,c) for (ll (a)=(b);(a)>=(c);--(a))
#define FORSQ(a,b,c) for (ll (a)=(b);(a)*(a)<=(c);++(a))
#define FORC(a,b,c) for (char (a)=(b);(a)<=(c);++(a))
#define FOREACH(a,b) for (auto &(a) : (b))
#define rep(i,n) FOR(i,0,n)
#define repn(i,n) FORN(i,1,n)
#define drep(i,n) DFOR(i,n-1,0)
#define drepn(i,n) DFOR(i,n,1)
#define MAX(a,b) a = Max(a,b)
#define MIN(a,b) a = Min(a,b)
#define SQR(x) ((LL)(x) * (x))
#define Reset(a,b) memset(a,b,sizeof(a))
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define all(v) v.begin(),v.end()
#define ALLA(arr,sz) arr,arr+sz
#define SIZE(v) (int)v.size()
#define SORT(v) sort(all(v))
#define REVERSE(v) reverse(ALL(v))
#define SORTA(arr,sz) sort(ALLA(arr,sz))
#define REVERSEA(arr,sz) reverse(ALLA(arr,sz))
#define PERMUTE next_permutation
#define TC(t) while(t--)
#define forever for(;;)
#define PINF 1000000000000
#define newline '
'

#define test if(1)if(0)cerr
using namespace std;
using namespace std;
typedef vector<int> vi;
typedef vector<vi> vvi;
typedef pair<int,int> ii;
typedef pair<double,double> dd;
typedef pair<char,char> cc;
typedef vector<ii> vii;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll, ll> l4;
const double pi = acos(-1.0);

const int maxnode = 4e5+5, sigma_size = 26;
struct Trie
{
    int g[maxnode][sigma_size], sz;
    int cnt[sigma_size];
    void init()
    {
        sz = 1; Reset(g[0], 0);
        Reset(cnt, 0);
    }
    int idx(char c)
    {
        return c - 'a';
    }
    void insert(const string &str)
    {
        int u = 0, n = str.length();
        rep(i, n)
        {
            int c = idx(str[i]);

            if (!g[u][c])
            {
                Reset(g[sz], 0);
                g[u][c] = sz++;
                if (i)  cnt[c] += 1;
            }
            u = g[u][c];
        }
    }
};
Trie prefix, suffix;
bitset<sigma_size> vis;
int n;

int main()
{
    
    ios::sync_with_stdio(false);
    cin.tie(0);
    while (cin >> n)
    {
        string line;
        prefix.init(), suffix.init();
        vis.reset();
        rep(i, n)
        {
            cin >> line;
            if (line.length() == 1) vis[line[0]-'a'] = true;
            prefix.insert(line);
            reverse(line.begin(), line.end());
            suffix.insert(line);
        }
        ll ans = (ll) (prefix.sz-1) * (suffix.sz-1);
        rep(i, sigma_size)
        {
            ans -= (ll) prefix.cnt[i] * suffix.cnt[i] - vis[i];
        }
        
        cout << ans << newline;
    }
    
    
    


}