結果
問題 | No.430 文字列検索 |
ユーザー |
|
提出日時 | 2024-12-15 20:44:06 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 8 ms / 2,000 ms |
コード長 | 24,270 bytes |
コンパイル時間 | 6,832 ms |
コンパイル使用メモリ | 229,764 KB |
実行使用メモリ | 6,656 KB |
最終ジャッジ日時 | 2024-12-15 20:44:14 |
合計ジャッジ時間 | 7,503 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 14 |
ソースコード
#define PROBLEM "https://yukicoder.me/problems/no/430"#ifndef STRING_AHO_CORASICK_HPP#define STRING_AHO_CORASICK_HPP 1#include <algorithm>#include <queue>#include <string>#include <type_traits>#include <unordered_map>#include <vector>#ifndef DATA_STRUCTURE_TRIE_HPP#define DATA_STRUCTURE_TRIE_HPP 1#include <cassert>#include <cstring>#include <functional>#include <string>#include <vector>namespace kk2 {template <int char_size> struct TrieNode {int nxt[char_size];// 通過した回数,終点は含まないint passing;// このノードが終点のidたちstd::vector<int> accept;TrieNode() : passing(0) { std::fill(nxt, nxt + char_size, -1); }};template <int char_size, int margin> struct Trie {using Node = TrieNode<char_size>;std::vector<Node> nodes;constexpr static int root = 0;Trie() {nodes.emplace_back();}Trie(int Num) {nodes.reserve(Num);nodes.emplace_back();}int push_node() {nodes.emplace_back();return (int)nodes.size() - 1;}void update_direct(int node, int id) { nodes[node].accept.push_back(id); }void update_child(int node) { ++nodes[node].passing; }void add(const std::string &str) {assert(!str.empty());const int id = nodes[root].passing;int now = root;for (int i = 0; i < (int)str.size(); ++i) {const int d = str[i] - margin;if (nodes[now].nxt[d] == -1) nodes[now].nxt[d] = push_node();update_child(now);now = nodes[now].nxt[d];}update_direct(now, id);}template <void (*f)(int)> void query(const std::string &str) {query(str, [](int idx) { f(idx); });}template <class F> void query(const std::string &str, const F &f) {int now = root;for (char c : str) {for (int &idx : nodes[now].accept) f(idx);const int d = c - margin;now = nodes[now].nxt[d];if (now == -1) return;}for (int idx : nodes[now].accept) f(idx);}int count() const { return nodes[0].passing; }int size() const { return (int)nodes.size(); }// return the number of strings which have the prefix corresponding to the node_idint size(int node_idx) const {return (int)nodes[node_idx].accept.size() + nodes[node_idx].passing;}};} // namespace kk2#endif // DATA_STRUCTURE_TRIE_HPP// #include "../data_structure/trie.hpp"namespace kk2 {template <int char_size, int margin> struct AhoCorasick : Trie<char_size + 1, margin> {using Trie<char_size + 1, margin>::Trie;using Trie<char_size + 1, margin>::count;constexpr static int FAIL = char_size;std::vector<int> correct, perm;void build() {correct.resize(this->size());int now = 0;perm.resize(this->size());perm[now++] = this->root;for (int i = 0; i < (int)this->size(); ++i) {correct[i] = (int)this->nodes[i].accept.size();}std::queue<int> que;for (int i = 0; i <= char_size; ++i) {if (this->nodes[this->root].nxt[i] == -1) {this->nodes[this->root].nxt[i] = this->root;} else {this->nodes[this->nodes[this->root].nxt[i]].nxt[FAIL] = this->root;que.emplace(this->nodes[this->root].nxt[i]);}}while (!que.empty()) {perm[now++] = que.front();auto &now = this->nodes[que.front()];int fail = now.nxt[FAIL];correct[que.front()] += correct[fail];que.pop();for (int i = 0; i < char_size; ++i) {if (now.nxt[i] == -1) {now.nxt[i] = this->nodes[fail].nxt[i];} else {this->nodes[now.nxt[i]].nxt[FAIL] = this->nodes[fail].nxt[i];que.emplace(now.nxt[i]);}}}}long long all_match(const std::string &str, int now_ = 0) {std::unordered_map<int, int> visit_cnt;for (char c : str) {now_ = this->nodes[now_].nxt[c - margin];visit_cnt[now_]++;}long long res{};for (auto &&[now, cnt] : visit_cnt) { res += (long long)correct[now] * cnt; }return res;}std::vector<long long> each_match(const std::string &str, int now_ = 0) {std::vector<int> visit_cnt(this->size());for (char c : str) {now_ = this->nodes[now_].nxt[c - margin];visit_cnt[now_]++;}std::vector<long long> res(this->count());for (int i = this->size() - 1; i > 0; --i) {int now = perm[i];visit_cnt[this->nodes[now].nxt[FAIL]] += visit_cnt[now];for (int idx : this->nodes[now].accept) { res[idx] += visit_cnt[now]; }}return res;}int move(int now, char c) { return this->nodes[now].nxt[c - margin]; }int count(int node) const { return correct[node]; }};} // namespace kk2#endif // STRING_AHO_CORASICK_HPP// #include "../../string/aho_corasick.hpp"#ifndef TEMPLATE#define TEMPLATE 1// #pragma GCC optimize("O3,unroll-loops")// #include <bits/stdc++.h>#include <algorithm>#include <array>#include <bitset>#include <cassert>#include <chrono>#include <cmath>#include <deque>#include <functional>#include <iterator>#include <limits>#include <map>#include <numeric>#include <optional>#include <queue>#include <random>#include <set>#include <stack>#include <string>#include <unordered_map>#include <unordered_set>#include <utility>#include <vector>#ifndef TEMPLATE_FASTIO_HPP#define TEMPLATE_FASTIO_HPP 1#include <cctype>#include <cstdint>#include <cstdio>#include <fstream>#include <iostream>#include <string>#ifndef TYPE_TRAITS_HPP#define TYPE_TRAITS_HPP 1#include <istream>#include <ostream>#include <type_traits>namespace kk2 {template <typename T>using is_signed_int128 = typename std::conditional<std::is_same<T, __int128_t>::valueor std::is_same<T, __int128>::value,std::true_type,std::false_type>::type;template <typename T>using is_unsigned_int128 =typename std::conditional<std::is_same<T, __uint128_t>::valueor std::is_same<T, unsigned __int128>::value,std::true_type,std::false_type>::type;template <typename T>using is_integral =typename std::conditional<std::is_integral<T>::value or is_signed_int128<T>::valueor is_unsigned_int128<T>::value,std::true_type,std::false_type>::type;template <typename T>using is_signed = typename std::conditional<std::is_signed<T>::value or is_signed_int128<T>::value,std::true_type,std::false_type>::type;template <typename T>using is_unsigned =typename std::conditional<std::is_unsigned<T>::value or is_unsigned_int128<T>::value,std::true_type,std::false_type>::type;template <typename T>using make_unsigned_int128 =typename std::conditional<std::is_same<T, __int128_t>::value, __uint128_t, unsigned __int128>;template <typename T>using to_unsigned =typename std::conditional<is_signed_int128<T>::value,make_unsigned_int128<T>,typename std::conditional<std::is_signed<T>::value,std::make_unsigned<T>,std::common_type<T>>::type>::type;template <typename T> using is_integral_t = std::enable_if_t<is_integral<T>::value>;template <typename T> using is_signed_t = std::enable_if_t<is_signed<T>::value>;template <typename T> using is_unsigned_t = std::enable_if_t<is_unsigned<T>::value>;template <typename T>using is_function_pointer =typename std::conditional<std::is_pointer_v<T> && std::is_function_v<std::remove_pointer_t<T>>,std::true_type,std::false_type>::type;template <typename T, std::enable_if_t<is_function_pointer<T>::value> * = nullptr>struct is_two_args_function_pointer : std::false_type {};template <typename R, typename T1, typename T2>struct is_two_args_function_pointer<R (*)(T1, T2)> : std::true_type {};template <typename T>using is_two_args_function_pointer_t = std::enable_if_t<is_two_args_function_pointer<T>::value>;namespace type_traits {struct istream_tag {};struct ostream_tag {};} // namespace type_traitstemplate <typename T>using is_standard_istream = typename std::conditional<std::is_same<T, std::istream>::value|| std::is_same<T, std::ifstream>::value,std::true_type,std::false_type>::type;template <typename T>using is_standard_ostream = typename std::conditional<std::is_same<T, std::ostream>::value|| std::is_same<T, std::ofstream>::value,std::true_type,std::false_type>::type;template <typename T> using is_user_defined_istream = std::is_base_of<type_traits::istream_tag, T>;template <typename T> using is_user_defined_ostream = std::is_base_of<type_traits::ostream_tag, T>;template <typename T>using is_istream =typename std::conditional<is_standard_istream<T>::value || is_user_defined_istream<T>::value,std::true_type,std::false_type>::type;template <typename T>using is_ostream =typename std::conditional<is_standard_ostream<T>::value || is_user_defined_ostream<T>::value,std::true_type,std::false_type>::type;template <typename T> using is_istream_t = std::enable_if_t<is_istream<T>::value>;template <typename T> using is_ostream_t = std::enable_if_t<is_ostream<T>::value>;} // namespace kk2#endif // TYPE_TRAITS_HPP// #include "../type_traits/type_traits.hpp"namespace kk2 {namespace fastio {#ifndef INPUT_FILE#define INPUT_FILE "in.txt"#endif#ifndef OUTPUT_FILE#define OUTPUT_FILE "out.txt"#endifstruct Scanner : type_traits::istream_tag {private:static constexpr size_t INPUT_BUF = 1 << 17;size_t pos = 0, end = 0;static char buf[INPUT_BUF];FILE *fp;public:Scanner() : fp(stdin) {}Scanner(const char *file) : fp(fopen(file, "r")) {}~Scanner() {if (fp != stdin) fclose(fp);}char now() {if (pos == end) {while (!(end = fread(buf, 1, INPUT_BUF, fp))) {}if (end != INPUT_BUF) buf[end] = '\0';pos = 0;}return buf[pos];}void skip_space() {while (isspace(now())) ++pos;}template <class T, is_unsigned_t<T> * = nullptr> T next_unsigned_integral() {skip_space();T res{};while (isdigit(now())) {res = res * 10 + (now() - '0');++pos;}return res;}template <class T, is_signed_t<T> * = nullptr> T next_signed_integral() {skip_space();if (now() == '-') {++pos;return T(-next_unsigned_integral<typename to_unsigned<T>::type>());} else return (T)next_unsigned_integral<typename to_unsigned<T>::type>();}char next_char() {skip_space();auto res = now();++pos;return res;}std::string next_string() {skip_space();std::string res;while (true) {char c = now();if (isspace(c) or c == '\0') break;res.push_back(now());++pos;}return res;}template <class T, is_unsigned_t<T> * = nullptr> Scanner &operator>>(T &x) {x = next_unsigned_integral<T>();return *this;}template <class T, is_signed_t<T> * = nullptr> Scanner &operator>>(T &x) {x = next_signed_integral<T>();return *this;}Scanner &operator>>(char &x) {x = next_char();return *this;}Scanner &operator>>(std::string &x) {x = next_string();return *this;}};struct endl_struct_t {};struct Printer : type_traits::ostream_tag {private:static char helper[10000][5];static char leading_zero[10000][5];constexpr static size_t OUTPUT_BUF = 1 << 17;static char buf[OUTPUT_BUF];size_t pos = 0;FILE *fp;template <class T> static constexpr void div_mod(T &a, T &b, T mod) {a = b / mod;b -= a * mod;}static void init() {buf[0] = '\0';for (size_t i = 0; i < 10000; ++i) {leading_zero[i][0] = i / 1000 + '0';leading_zero[i][1] = i / 100 % 10 + '0';leading_zero[i][2] = i / 10 % 10 + '0';leading_zero[i][3] = i % 10 + '0';leading_zero[i][4] = '\0';size_t j = 0;if (i >= 1000) helper[i][j++] = i / 1000 + '0';if (i >= 100) helper[i][j++] = i / 100 % 10 + '0';if (i >= 10) helper[i][j++] = i / 10 % 10 + '0';helper[i][j++] = i % 10 + '0';helper[i][j] = '\0';}}public:Printer() : fp(stdout) { init(); }Printer(const char *file) : fp(fopen(file, "w")) { init(); }~Printer() {write();if (fp != stdout) fclose(fp);}void write() {fwrite(buf, 1, pos, fp);pos = 0;}void flush() {write();fflush(fp);}void put_char(char c) {if (pos == OUTPUT_BUF) write();buf[pos++] = c;}void put_cstr(const char *s) {while (*s) put_char(*(s++));}void put_u32(uint32_t x) {uint32_t y;if (x >= 100000000) { // 10^8div_mod<uint32_t>(y, x, 100000000);put_cstr(helper[y]);div_mod<uint32_t>(y, x, 10000);put_cstr(leading_zero[y]);put_cstr(leading_zero[x]);} else if (x >= 10000) { // 10^4div_mod<uint32_t>(y, x, 10000);put_cstr(helper[y]);put_cstr(leading_zero[x]);} else put_cstr(helper[x]);}void put_i32(int32_t x) {if (x < 0) {put_char('-');put_u32(-x);} else put_u32(x);}void put_u64(uint64_t x) {uint64_t y;if (x >= 1000000000000ull) { // 10^12div_mod<uint64_t>(y, x, 1000000000000ull);put_u32(y);div_mod<uint64_t>(y, x, 100000000ull);put_cstr(leading_zero[y]);div_mod<uint64_t>(y, x, 10000ull);put_cstr(leading_zero[y]);put_cstr(leading_zero[x]);} else if (x >= 10000ull) { // 10^4div_mod<uint64_t>(y, x, 10000ull);put_u32(y);put_cstr(leading_zero[x]);} else put_cstr(helper[x]);}void put_i64(int64_t x) {if (x < 0) {put_char('-');put_u64(-x);} else put_u64(x);}void put_u128(__uint128_t x) {constexpr static __uint128_t pow10_10 = 10000000000ull;constexpr static __uint128_t pow10_20 = pow10_10 * pow10_10;__uint128_t y;if (x >= pow10_20) { // 10^20div_mod<__uint128_t>(y, x, pow10_20);put_u64(uint64_t(y));div_mod<__uint128_t>(y, x, __uint128_t(10000000000000000ull));put_cstr(leading_zero[y]);div_mod<__uint128_t>(y, x, __uint128_t(1000000000000ull));put_cstr(leading_zero[y]);div_mod<__uint128_t>(y, x, __uint128_t(100000000ull));put_cstr(leading_zero[y]);div_mod<__uint128_t>(y, x, __uint128_t(10000ull));put_cstr(leading_zero[y]);put_cstr(leading_zero[x]);} else if (x >= __uint128_t(10000)) { // 10^4div_mod<__uint128_t>(y, x, __uint128_t(10000));put_u64(uint64_t(y));put_cstr(leading_zero[x]);} else put_cstr(helper[x]);}void put_i128(__int128_t x) {if (x < 0) {put_char('-');put_u128(-x);} else put_u128(x);}template <class T, is_unsigned_t<T> * = nullptr> Printer &operator<<(T x) {if constexpr (sizeof(T) <= 4) put_u32(x);else if constexpr (sizeof(T) <= 8) put_u64(x);else put_u128(x);return *this;}template <class T, is_signed_t<T> * = nullptr> Printer &operator<<(T x) {if constexpr (sizeof(T) <= 4) put_i32(x);else if constexpr (sizeof(T) <= 8) put_i64(x);else put_i128(x);return *this;}Printer &operator<<(char x) {put_char(x);return *this;}Printer &operator<<(const std::string &x) {for (char c : x) put_char(c);return *this;}Printer &operator<<(const char *x) {put_cstr(x);return *this;}// std::cout << std::endl; は関数ポインタを渡しているらしいPrinter &operator<<(endl_struct_t) {put_char('\n');flush();return *this;}};char Scanner::buf[Scanner::INPUT_BUF];char Printer::buf[Printer::OUTPUT_BUF];char Printer::helper[10000][5];char Printer::leading_zero[10000][5];} // namespace fastio#if defined(INTERACTIVE) || defined(USE_STDIO)#ifdef KK2std::ifstream kin(INPUT_FILE);std::ofstream kout(OUTPUT_FILE);auto (*kendl)(std::ostream &) = std::endl<char, std::char_traits<char>>;#elseauto &kin = std::cin;auto &kout = std::cout;auto (*kendl)(std::ostream &) = std::endl<char, std::char_traits<char>>;#endif#elif defined(KK2)fastio::Scanner kin(INPUT_FILE);fastio::Printer kout(OUTPUT_FILE);fastio::endl_struct_t kendl;#elsefastio::Scanner kin;fastio::Printer kout;fastio::endl_struct_t kendl;#endif} // namespace kk2#endif // TEMPLATE_FASTIO_HPP// #include "fastio.hpp"#ifndef TEMPLATE_TYPE_ALIAS_HPP#define TEMPLATE_TYPE_ALIAS_HPP 1#include <utility>#include <vector>#include <queue>#include <functional>using u32 = unsigned int;using i64 = long long;using u64 = unsigned long long;using i128 = __int128_t;using u128 = __uint128_t;using pi = std::pair<int, int>;using pl = std::pair<i64, i64>;using pil = std::pair<int, i64>;using pli = std::pair<i64, int>;template <class T> using vc = std::vector<T>;template <class T> using vvc = std::vector<vc<T>>;template <class T> using vvvc = std::vector<vvc<T>>;template <class T> using vvvvc = std::vector<vvvc<T>>;template <class T> using pq = std::priority_queue<T>;template <class T> using pqi = std::priority_queue<T, std::vector<T>, std::greater<T>>;#endif // TEMPLATE_TYPE_ALIAS_HPP// #include "type_alias.hpp"#ifndef TEMPLATE_CONSTANT_HPP#define TEMPLATE_CONSTANT_HPP 1template <class T> constexpr T infty = 0;template <> constexpr int infty<int> = (1 << 30) - 123;template <> constexpr i64 infty<i64> = (1ll << 62) - (1ll << 31);template <> constexpr i128 infty<i128> = (i128(1) << 126) - (i128(1) << 63);template <> constexpr u32 infty<u32> = infty<int>;template <> constexpr u64 infty<u64> = infty<i64>;template <> constexpr u128 infty<u128> = infty<i128>;template <> constexpr double infty<double> = infty<i64>;template <> constexpr long double infty<long double> = infty<i64>;constexpr int mod = 998244353;constexpr int modu = 1e9 + 7;constexpr long double PI = 3.14159265358979323846;#endif // TEMPLATE_CONSTANT_HPP// #include "constant.hpp"#ifndef TEMPLATE_FUNCTIONAL_UTIL_HPP#define TEMPLATE_FUNCTIONAL_UTIL_HPP 1#include <algorithm>#include <vector>namespace kk2 {template <class T, class... Sizes> auto make_vector(int first, Sizes... sizes) {if constexpr (sizeof...(sizes) == 0) {return std::vector<T>(first);} else {return std::vector<decltype(make_vector<T>(sizes...))>(first, make_vector<T>(sizes...));}}template <class T, class U> void fill_all(std::vector<T> &v, const U &x) {std::fill(std::begin(v), std::end(v), T(x));}template <class T, class U> void fill_all(std::vector<std::vector<T>> &v, const U &x) {for (auto &u : v) fill_all(u, x);}} // namespace kk2template <class T, class S> inline bool chmax(T &a, const S &b) {return (a < b ? a = b, 1 : 0);}template <class T, class S> inline bool chmin(T &a, const S &b) {return (a > b ? a = b, 1 : 0);}#endif // TEMPLATE_FUNCTIONAL_UTIL_HPP// #include "function_util.hpp"#ifndef TEMPLATE_MACROS_HPP#define TEMPLATE_MACROS_HPP 1#define rep1(a) for (long long _ = 0; _ < (long long)(a); ++_)#define rep2(i, a) for (long long i = 0; i < (long long)(a); ++i)#define rep3(i, a, b) for (long long i = (a); i < (long long)(b); ++i)#define repi2(i, a) for (long long i = (a) - 1; i >= 0; --i)#define repi3(i, a, b) for (long long i = (a) - 1; i >= (long long)(b); --i)#define overload3(a, b, c, d, ...) d#define rep(...) overload3(__VA_ARGS__, rep3, rep2, rep1)(__VA_ARGS__)#define repi(...) overload3(__VA_ARGS__, repi3, repi2, rep1)(__VA_ARGS__)#define fi first#define se second#define all(p) p.begin(), p.end()#endif // TEMPLATE_MACROS_HPP// #include "macros.hpp"#ifndef TEMPLATE_IO_UTIL_HPP#define TEMPLATE_IO_UTIL_HPP 1#include <array>#include <utility>#include <vector>// #include "../type_traits/type_traits.hpp"template <class IStream, class T, class U, kk2::is_istream_t<IStream> * = nullptr>IStream &operator>>(IStream &is, std::pair<T, U> &p) {is >> p.first >> p.second;return is;}template <class OStream, class T, class U, kk2::is_ostream_t<OStream> * = nullptr>OStream &operator<<(OStream &os, const std::pair<T, U> &p) {os << p.first << ' ' << p.second;return os;}template <class IStream, class T, kk2::is_istream_t<IStream> * = nullptr>IStream &operator>>(IStream &is, std::vector<T> &v) {for (auto &x : v) is >> x;return is;}template <class OStream, class T, kk2::is_ostream_t<OStream> * = nullptr>OStream &operator<<(OStream &os, const std::vector<T> &v) {for (int i = 0; i < (int)v.size(); i++) { os << v[i] << (i + 1 == (int)v.size() ? "" : " "); }return os;}template <class IStream, class T, size_t F, kk2::is_istream_t<IStream> * = nullptr>IStream &operator>>(IStream &is, std::array<T, F> &a) {for (auto &x : a) is >> x;return is;}template <class OStream, class T, size_t F, kk2::is_ostream_t<OStream> * = nullptr>OStream &operator<<(OStream &os, const std::array<T, F> &a) {for (int i = 0; i < (int)F; i++) { os << a[i] << (i + 1 == (int)F ? "" : " "); }return os;}#endif // TEMPLATE_IO_UTIL_HPP// #include "io_util.hpp"using kk2::kendl;using kk2::kin;using kk2::kout;void Yes(bool b = 1) {kout << (b ? "Yes\n" : "No\n");}void No(bool b = 1) {kout << (b ? "No\n" : "Yes\n");}void YES(bool b = 1) {kout << (b ? "YES\n" : "NO\n");}void NO(bool b = 1) {kout << (b ? "NO\n" : "YES\n");}void yes(bool b = 1) {kout << (b ? "yes\n" : "no\n");}void no(bool b = 1) {kout << (b ? "no\n" : "yes\n");}#endif // TEMPLATE// #include "../../template/template.hpp"using namespace std;int main() {string s;kin >> s;int m;kin >> m;vc<string> c(m);kin >> c;kk2::AhoCorasick<26, 'A'> ac(1e5);for (auto &x : c) ac.add(x);ac.build();auto each = ac.each_match(s);i64 res = accumulate(all(each), 0LL);kout << res << kendl;return 0;}// converted!!