hihoCoder#1036 Trie图

原题地址

看了这篇博文，总算是把Trie图弄明白了

Runtime Error了无数次，一直不知道为什么，于是写了个脚本生成了一组大数据，发现果然段错误了。

调试了一下午，总算闹明白了，为什么呢？

1. 空间超大的变量不要放在函数里，会爆栈，应该弄成全局变量或者是从堆上动态分配。

2. 看清题目的数据范围，一开始我的MAX_NODE设的是1024。。。

代码：

 1 #include <iostream>
 2 #include <cstring>
 3
 4 using namespace std;
 5
 6 #define MAX_NODE 1000010
 7 #define SIGMA_SIZE 32
 8
 9 int q[MAX_NODE];
10
11 struct TrieGraph {
12   int f[MAX_NODE];
13   int g[MAX_NODE][SIGMA_SIZE];
14   int m[MAX_NODE];
15   int size;
16
17   void init() {
18     size = 1;
19     memset(f, 0, sizeof(f));
20     memset(g[0], 0, sizeof(g[0]));
21   }
22
23   int index(char c) {
24     return c - ‘a‘;
25   }
26
27   void insert(const char *s) {
28     int u = 0;
29     while (*s) {
30       int i = index(*s);
31       if (!g[u][i]) {
32         memset(g[size], 0, sizeof(g[size]));
33         m[size] = 0;
34         g[u][i] = size++;
35       }
36       u = g[u][i];
37       s++;
38     }
39     m[u] = 1;
40   }
41
42   void build() {
43     int qh = 0, qt = 0;
44     f[0] = 0;
45     for (int i = 0; i < 26; i++) {
46       int &p = g[0][i];
47       if (p) {
48         f[p] = 0;
49         q[qt++] = p;
50       }
51       else
52         p = 0;
53     }
54     while (qh < qt) {
55       int u = q[qh++];
56       for (int i = 0; i < 26; i++) {
57         int &v = g[u][i];
58         if (v) {
59           q[qt++] = v;
60           f[v] = g[f[u]][i];
61           m[u] |= m[f[u]];
62         }
63         else
64           v = g[f[u]][i];
65       }
66     }
67   }
68
69   bool find(const char *s) {
70     int u = 0;
71     while (*s) {
72       int i = index(*s);
73       while (u && !g[u][i])
74         u = f[u];
75       u = g[u][i];
76       if (m[u])
77         return true;
78       s++;
79     }
80     return false;
81   }
82 } tg;
83
84 int main() {
85   int N;
86   string s;
87
88   tg.init();
89   cin >> N;
90   for (int i = 0; i < N; i++) {
91     cin >> s;
92     tg.insert(s.c_str());
93   }
94   tg.build();
95   cin >> s;
96   cout << (tg.find(s.c_str()) ? "YES" : "NO") << endl;
97   return 0;
98 }