#line 1 "a.cpp" #define PROBLEM "" #line 2 "/home/kuhaku/home/github/algo/lib/template/template.hpp" #pragma GCC target("sse4.2,avx2,bmi2") #pragma GCC optimize("O3") #pragma GCC optimize("unroll-loops") #include using namespace std; template bool chmax(T &a, const U &b) { return a < (T)b ? a = (T)b, true : false; } template bool chmin(T &a, const U &b) { return (T)b < a ? a = (T)b, true : false; } constexpr int64_t INF = 1000000000000000003; constexpr int Inf = 1000000003; constexpr int MOD = 1000000007; constexpr int MOD_N = 998244353; constexpr double EPS = 1e-7; constexpr double PI = M_PI; #line 2 "/home/kuhaku/home/github/algo/lib/string/aho_corasick.hpp" template struct aho_corasick { private: struct _node { std::vector next_node; _node() : next_node(char_size, -1) {} int next(int x) const { return this->next_node[x]; } }; public: using node_type = _node; aho_corasick() : nodes() { this->nodes.emplace_back(); } int size() const noexcept { return this->nodes.size(); } std::vector build() { this->failure = std::vector(nodes.size(), 0); queue que; for (int i = 0; i < char_size; ++i) { int next_root = this->nodes[0].next(i); if (next_root != -1) { que.emplace(next_root); } } while (!que.empty()) { int x = que.front(); que.pop(); for (int i = 0; i < char_size; ++i) { int next_x = this->nodes[x].next(i); if (next_x != -1) { que.emplace(next_x); int y = x; do { y = this->failure[y]; int next_y = this->nodes[y].next(i); if (next_y != -1) { this->failure[next_x] = next_y; break; } } while (y != 0); } } } return this->failure; } std::vector insert(const string &word) { std::vector res; int node_id = 0; for (int i = 0; i < (int)word.size(); ++i) { int c = word[i] - base; int &next_id = this->nodes[node_id].next_node[c]; if (next_id == -1) { next_id = this->nodes.size(); this->nodes.emplace_back(); } node_id = next_id; res.emplace_back(node_id); } return res; } int search(const char c, int now = 0) { int next_id = this->nodes[now].next(c - base); while (next_id == -1 && now != 0) { now = this->failure[now]; next_id = this->nodes[now].next(c - base); } return next_id != -1 ? next_id : 0; } std::vector search(const std::string &str, int now = 0) { std::vector res; res.emplace_back(now); for (auto c : str) { now = search(c, now); res.emplace_back(now); } return res; } node_type get_node(int node_id) const { assert(0 <= node_id && node_id < (int)this->nodes.size()); return this->nodes[node_id]; } private: std::vector nodes; std::vector failure; }; #line 3 "/home/kuhaku/home/github/algo/lib/template/macro.hpp" #define FOR(i, m, n) for (int i = (m); i < int(n); ++i) #define FORR(i, m, n) for (int i = (m)-1; i >= int(n); --i) #define FORL(i, m, n) for (int64_t i = (m); i < int64_t(n); ++i) #define rep(i, n) FOR (i, 0, n) #define repn(i, n) FOR (i, 1, n + 1) #define repr(i, n) FORR (i, n, 0) #define repnr(i, n) FORR (i, n + 1, 1) #define all(s) (s).begin(), (s).end() #line 3 "/home/kuhaku/home/github/algo/lib/template/sonic.hpp" struct Sonic { Sonic() { std::ios::sync_with_stdio(false); std::cin.tie(nullptr); } constexpr void operator()() const {} } sonic; #line 5 "/home/kuhaku/home/github/algo/lib/template/atcoder.hpp" using ll = int64_t; using ld = long double; template std::istream &operator>>(std::istream &is, std::pair &p) { return is >> p.first >> p.second; } template std::istream &operator>>(std::istream &is, std::vector &v) { for (T &i : v) is >> i; return is; } template std::ostream &operator<<(std::ostream &os, const std::pair &p) { return os << '(' << p.first << ',' << p.second << ')'; } template std::ostream &operator<<(std::ostream &os, const std::vector &v) { for (auto it = v.begin(); it != v.end(); ++it) { os << (it == v.begin() ? "" : " ") << *it; } return os; } template void co(Head &&head, Tail &&...tail) { if constexpr (sizeof...(tail) == 0) std::cout << head << '\n'; else std::cout << head << ' ', co(std::forward(tail)...); } template void ce(Head &&head, Tail &&...tail) { if constexpr (sizeof...(tail) == 0) std::cerr << head << '\n'; else std::cerr << head << ' ', ce(std::forward(tail)...); } template auto make_vector(T x, int arg, Args... args) { if constexpr (sizeof...(args) == 0) return std::vector(arg, x); else return std::vector(arg, make_vector(x, args...)); } void setp(int n) { std::cout << std::fixed << std::setprecision(n); } void Yes(bool is_correct = true) { std::cout << (is_correct ? "Yes" : "No") << '\n'; } void No(bool is_not_correct = true) { Yes(!is_not_correct); } void YES(bool is_correct = true) { std::cout << (is_correct ? "YES" : "NO") << '\n'; } void NO(bool is_not_correct = true) { YES(!is_not_correct); } void Takahashi(bool is_correct = true) { std::cout << (is_correct ? "Takahashi" : "Aoki") << '\n'; } void Aoki(bool is_not_correct = true) { Takahashi(!is_not_correct); } #line 3 "/home/kuhaku/home/github/algo/lib/graph/graph.hpp" /** * @brief 重み付きグラフ * * @tparam T 辺の重みの型 */ template struct Graph { private: struct _edge { constexpr _edge() : _from(), _to(), _weight() {} constexpr _edge(int from, int to, T weight) : _from(from), _to(to), _weight(weight) {} constexpr bool operator<(const _edge &rhs) const { return this->weight() < rhs.weight(); } constexpr bool operator>(const _edge &rhs) const { return rhs < *this; } constexpr int from() const { return this->_from; } constexpr int to() const { return this->_to; } constexpr T weight() const { return this->_weight; } private: int _from, _to; T _weight; }; public: using edge_type = typename Graph::_edge; Graph(int v) : _size(v), edges(v) {} const auto &operator[](int i) const { return this->edges[i]; } auto &operator[](int i) { return this->edges[i]; } const auto begin() const { return this->edges.begin(); } auto begin() { return this->edges.begin(); } const auto end() const { return this->edges.end(); } auto end() { return this->edges.end(); } constexpr int size() const { return this->_size; } void add_edge(const edge_type &e) { this->edges[e.from()].emplace_back(e); } void add_edge(int from, int to, T weight = T(1)) { this->edges[from].emplace_back(from, to, weight); } void add_edges(int from, int to, T weight = T(1)) { this->edges[from].emplace_back(from, to, weight); this->edges[to].emplace_back(to, from, weight); } void input_edge(int m, int base = 1) { for (int i = 0; i < m; ++i) { int from, to; T weight; cin >> from >> to >> weight; this->add_edge(from - base, to - base, weight); } } void input_edges(int m, int base = 1) { for (int i = 0; i < m; ++i) { int from, to; T weight; cin >> from >> to >> weight; this->add_edges(from - base, to - base, weight); } } private: int _size; std::vector> edges; }; template <> struct Graph { private: struct _edge { constexpr _edge() : _from(), _to() {} constexpr _edge(int from, int to) : _from(from), _to(to) {} constexpr int from() const { return this->_from; } constexpr int to() const { return this->_to; } constexpr int weight() const { return 1; } constexpr bool operator<(const _edge &rhs) const { return this->weight() < rhs.weight(); } constexpr bool operator>(const _edge &rhs) const { return rhs < *this; } private: int _from, _to; }; public: using edge_type = typename Graph::_edge; Graph(int v) : _size(v), edges(v) {} const auto &operator[](int i) const { return this->edges[i]; } auto &operator[](int i) { return this->edges[i]; } const auto begin() const { return this->edges.begin(); } auto begin() { return this->edges.begin(); } const auto end() const { return this->edges.end(); } auto end() { return this->edges.end(); } constexpr int size() const { return this->_size; } void add_edge(const edge_type &e) { this->edges[e.from()].emplace_back(e); } void add_edge(int from, int to) { this->edges[from].emplace_back(from, to); } void add_edges(int from, int to) { this->edges[from].emplace_back(from, to); this->edges[to].emplace_back(to, from); } void input_edge(int m, int base = 1) { for (int i = 0; i < m; ++i) { int from, to; cin >> from >> to; this->add_edge(from - base, to - base); } } void input_edges(int m, int base = 1) { for (int i = 0; i < m; ++i) { int from, to; cin >> from >> to; this->add_edges(from - base, to - base); } } private: int _size; std::vector> edges; }; #line 3 "/home/kuhaku/home/github/algo/lib/tree/tree_function.hpp" template std::vector tree_bfs(const Graph &g, int r = 0) { int pos = 0; std::vector res; std::vector visited(g.size()); res.emplace_back(r); visited[r] = true; while (pos < res.size()) { auto index = res[pos++]; for (auto &e : g[index]) { if (visited[e.to()]) continue; res.emplace_back(e.to()); visited[e.to()] = true; } } return res; } std::vector tree_bfs(const std::vector &parents) { int n = parents.size(); Graph g(n); int r = 0; for (int i = 0; i < n; ++i) { if (parents[i] == -1 || parents[i] == i) { r = i; continue; } g.add_edges(i, parents[i]); } return tree_bfs(g, r); } template std::vector tree_dfs(const Graph &g, int r = 0) { std::vector res; auto dfs = [&g, &res](auto self, int index, int parent) { res.emplace_back(index); for (auto &e : g[index]) { if (e.to() == parent) continue; self(self, e.to(), index); } }; dfs(dfs, r, -1); return res; } /** * @brief 木の距離を求める * * @tparam T 辺の重みの型 * @tparam U 距離の型 * @param g 木 * @param r 根 * @return std::vector 各頂点の根からの距離 */ template std::vector tree_dist(const Graph &g, int r = 0) { std::vector res(g.size(), -1); std::stack st; res[r] = 0; st.emplace(r); while (!st.empty()) { auto index = st.top(); st.pop(); for (auto &e : g[index]) { if (res[e.to()] != -1) continue; res[e.to()] = res[index] + e.weight(); st.emplace(e.to()); } } return res; } /** * @brief 木の頂点の親を求める * * @tparam T 辺の重みの型 * @param g 木 * @param r 根 * @return std::vector */ template std::vector tree_parent(const Graph &g, int r = 0) { std::vector res(g.size(), -1); std::stack st; res[r] = r; st.emplace(r); while (!st.empty()) { auto index = st.top(); st.pop(); for (auto &e : g[index]) { if (res[e.to()] != -1) continue; res[e.to()] = index; st.emplace(e.to()); } } res[r] = -1; return res; } /** * @brief 部分木の大きさを求める * * @tparam T 辺の重みの型 * @param g グラフ * @param r 根 * @return std::vector */ template std::vector tree_subtree(const Graph &g, int r = 0) { std::vector res(g.size()); auto dfs = [&g, &res](auto self, int index) { res[index] = 1; for (auto &e : g[index]) { if (res[e.to()] != 0) continue; res[index] += self(self, e.to()); } return res[index]; }; dfs(dfs, r); return res; } #line 5 "a.cpp" int main(void) { string s; cin >> s; int m; cin >> m; aho_corasick<26, 'A'> aho; vector correct; rep (i, m) { string t; cin >> t; auto v = aho.insert(t); correct.resize(aho.size()); ++correct[v.back()]; } auto failure = aho.build(); auto bfs = tree_bfs(failure); for (auto x : bfs) correct[x] += correct[failure[x]]; auto res = aho.search(s); ll ans = 0; for (auto x : res) ans += correct[x]; co(ans); return 0; }