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:
1 #include <cstdio> 2 #include <cstring> 3 #include <algorithm> 4 #include <iostream> 5 #include <string> 6 #include <vector> 7 #include <stack> 8 #include <bitset> 9 #include <cstdlib> 10 #include <cmath> 11 #include <set> 12 #include <list> 13 #include <deque> 14 #include <map> 15 #include <queue> 16 #include <fstream> 17 #include <cassert> 18 #include <unordered_map> 19 #include <unordered_set> 20 #include <cmath> 21 #include <sstream> 22 #include <time.h> 23 #include <complex> 24 #include <iomanip> 25 #define Max(a,b) ((a)>(b)?(a):(b)) 26 #define Min(a,b) ((a)<(b)?(a):(b)) 27 #define FOR(a,b,c) for (ll (a)=(b);(a)<(c);++(a)) 28 #define FORN(a,b,c) for (ll (a)=(b);(a)<=(c);++(a)) 29 #define DFOR(a,b,c) for (ll (a)=(b);(a)>=(c);--(a)) 30 #define FORSQ(a,b,c) for (ll (a)=(b);(a)*(a)<=(c);++(a)) 31 #define FORC(a,b,c) for (char (a)=(b);(a)<=(c);++(a)) 32 #define FOREACH(a,b) for (auto &(a) : (b)) 33 #define rep(i,n) FOR(i,0,n) 34 #define repn(i,n) FORN(i,1,n) 35 #define drep(i,n) DFOR(i,n-1,0) 36 #define drepn(i,n) DFOR(i,n,1) 37 #define MAX(a,b) a = Max(a,b) 38 #define MIN(a,b) a = Min(a,b) 39 #define SQR(x) ((LL)(x) * (x)) 40 #define Reset(a,b) memset(a,b,sizeof(a)) 41 #define fi first 42 #define se second 43 #define mp make_pair 44 #define pb push_back 45 #define all(v) v.begin(),v.end() 46 #define ALLA(arr,sz) arr,arr+sz 47 #define SIZE(v) (int)v.size() 48 #define SORT(v) sort(all(v)) 49 #define REVERSE(v) reverse(ALL(v)) 50 #define SORTA(arr,sz) sort(ALLA(arr,sz)) 51 #define REVERSEA(arr,sz) reverse(ALLA(arr,sz)) 52 #define PERMUTE next_permutation 53 #define TC(t) while(t--) 54 #define forever for(;;) 55 #define PINF 1000000000000 56 #define newline '\n' 57 58 #define test if(1)if(0)cerr 59 using namespace std; 60 using namespace std; 61 typedef vector<int> vi; 62 typedef vector<vi> vvi; 63 typedef pair<int,int> ii; 64 typedef pair<double,double> dd; 65 typedef pair<char,char> cc; 66 typedef vector<ii> vii; 67 typedef long long ll; 68 typedef unsigned long long ull; 69 typedef pair<ll, ll> l4; 70 const double pi = acos(-1.0); 71 72 const int maxnode = 4e5+5, sigma_size = 26; 73 struct Trie 74 { 75 int g[maxnode][sigma_size], sz; 76 int cnt[sigma_size]; 77 void init() 78 { 79 sz = 1; Reset(g[0], 0); 80 Reset(cnt, 0); 81 } 82 int idx(char c) 83 { 84 return c - 'a'; 85 } 86 void insert(const string &str) 87 { 88 int u = 0, n = str.length(); 89 rep(i, n) 90 { 91 int c = idx(str[i]); 92 93 if (!g[u][c]) 94 { 95 Reset(g[sz], 0); 96 g[u][c] = sz++; 97 if (i) cnt[c] += 1; 98 } 99 u = g[u][c]; 100 } 101 } 102 }; 103 Trie prefix, suffix; 104 bitset<sigma_size> vis; 105 int n; 106 107 int main() 108 { 109 110 ios::sync_with_stdio(false); 111 cin.tie(0); 112 while (cin >> n) 113 { 114 string line; 115 prefix.init(), suffix.init(); 116 vis.reset(); 117 rep(i, n) 118 { 119 cin >> line; 120 if (line.length() == 1) vis[line[0]-'a'] = true; 121 prefix.insert(line); 122 reverse(line.begin(), line.end()); 123 suffix.insert(line); 124 } 125 ll ans = (ll) (prefix.sz-1) * (suffix.sz-1); 126 rep(i, sigma_size) 127 { 128 ans -= (ll) prefix.cnt[i] * suffix.cnt[i] - vis[i]; 129 } 130 131 cout << ans << newline; 132 } 133 134 135 136 137 138 }
