#406034 (C++17) No.430 文字列検索

• 問題: No.430 文字列検索
• ユーザー: cutmdo
• 提出日時: 2019-12-07 20:34:27
• 言語: C++17 (gcc 13.3.0 + boost 1.87.0)
• 結果: CE (最新)
  AC
  (最初)
• 実行時間: -
• コード長: 5,001 bytes
• コンパイル時間: 910 ms
• コンパイル使用メモリ: 109,968 KB
• 最終ジャッジ日時: 2025-01-08 09:36:28
• ジャッジサーバーID （参考情報）: judge2 / judge4

コンパイルメッセージ

main.cpp:150:20: error: ‘unique_ptr’ is not a member of ‘std’
  150 |   std::vector<std::unique_ptr<TrieTree>> m_next;
      |                    ^~~~~~~~~~
main.cpp:16:1: note: ‘std::unique_ptr’ is defined in header ‘<memory>’; did you forget to ‘#include <memory>’?
   15 | #include <unordered_set>
  +++ |+#include <memory>
   16 | #include <vector>
main.cpp:150:31: error: template argument 1 is invalid
  150 |   std::vector<std::unique_ptr<TrieTree>> m_next;
      |                               ^~~~~~~~
main.cpp:150:31: error: template argument 2 is invalid
main.cpp:150:39: error: expected unqualified-id before ‘>’ token
  150 |   std::vector<std::unique_ptr<TrieTree>> m_next;
      |                                       ^~
main.cpp: In member function ‘auto TrieTree<Val, ignore>::emplace(const std::vector<long long int>&, Val, ll)’:
main.cpp:158:10: error: ‘m_next’ was not declared in this scope
  158 |     if (!m_next[vec[it]]) {
      |          ^~~~~~
main.cpp:159:30: error: ‘make_unique’ is not a member of ‘std’
  159 |       m_next[vec[it]] = std::make_unique<TrieTree>();
      |                              ^~~~~~~~~~~
main.cpp:159:30: note: ‘std::make_unique’ is defined in header ‘<memory>’; did you forget to ‘#include <memory>’?
main.cpp:159:50: error: expected primary-expression before ‘>’ token
  159 |       m_next[vec[it]] = std::make_unique<TrieTree>();
      |                                                  ^
main.cpp:159:52: error: expected primary-expression before ‘)’ token
  159 |       m_next[vec[it]] = std::make_unique<TrieTree>();
      |                                                    ^
main.cpp:161:5: error: ‘m_next’ was not declared in this scope
  161 |     m_next[vec[it]]->emplace(vec, val, it + 1);
      |     ^~~~~~
main.cpp: In member function ‘auto TrieTree<Val, ignore>::find(const std::vector<long long int>&, ll, const Lambda&) const’:
main.cpp:172:10: error: �

ソースコード

#include <algorithm>
#include <bitset>
#include <cassert>
#include <cmath>
#include <iomanip>
#include <iostream>
#include <list>
#include <map>
#include <numeric>
#include <queue>
#include <set>
#include <stack>
#include <string>
#include <unordered_map>
#include <unordered_set>
#include <vector>
#ifdef DEBUG
#include "./CompetitiveProgrammingCpp/Timer.hpp"
#include "./CompetitiveProgrammingCpp/debug.hpp"
#include "./CompetitiveProgrammingCpp/sample.hpp"
#else
#define dump(...)
template <class T>
inline auto d_val(T a, T b) {
  return a;
}
#endif

/* (=＾o＾=) */
#define int ll

/* macro */
#define FOR(i, b, e) for (ll i = (ll)(b); i < (ll)(e); ++i)
#define RFOR(i, b, e) for (ll i = (ll)(e - 1); i >= (ll)(b); --i)
#define REP(i, n) FOR(i, 0, n)
#define RREP(i, n) RFOR(i, 0, n)
#define REPC(x, c) for (const auto& x : (c))
#define REPI2(it, b, e) for (auto it = (b); it != (e); ++it)
#define REPI(it, c) REPI2(it, (c).begin(), (c).end())
#define RREPI(it, c) REPI2(it, (c).rbegin(), (c).rend())
#define REPI_ERACE2(it, b, e) for (auto it = (b); it != (e);)
#define REPI_ERACE(it, c) REPI_ERACE2(it, (c).begin(), (c).end())
#define ALL(x) (x).begin(), (x).end()
#define cauto const auto&
/* macro func */
template <class T>
inline auto sort(T& t) {
  std::sort(ALL(t));
}
template <class T>
inline auto rsort(T& t) {
  std::sort((t).rbegin(), (t).rend());
}
template <class T>
inline auto unique(T& t) {
  (t).erase(unique((t).begin(), (t).end()), (t).end());
}
template <class T, class S>
inline auto chmax(T& t, const S& s) {
  if (s > t) {
    t = s;
    return true;
  }
  return false;
}
template <class T, class S>
inline auto chmin(T& t, const S& s) {
  if (s < t) {
    t = s;
    return true;
  }
  return false;
}
inline auto BR() { std::cout << "\n"; }

/* type define */
using ll = long long;
using PAIR = std::pair<ll, ll>;
using VS = std::vector<std::string>;
using VL = std::vector<long long>;
using VVL = std::vector<VL>;
using VVVL = std::vector<VVL>;
using VD = std::vector<double>;
template <class T>
using V = std::vector<T>;

/* using std */
using std::cout;
constexpr char endl = '\n';
using std::bitset;
using std::cin;
using std::list;
using std::map;
using std::multimap;
using std::multiset;
using std::pair;
using std::priority_queue;
using std::queue;
using std::set;
using std::stack;
using std::string;
using std::unordered_map;
using std::unordered_multimap;
using std::unordered_multiset;
using std::unordered_set;
using std::vector;

/* constant value */
constexpr ll MOD = 1000000007;
// constexpr ll MOD = 998244353;

/* Initial processing  */
struct Preprocessing {
  Preprocessing() {
    std::cin.tie(0);
    std::ios::sync_with_stdio(0);
  };
} _Preprocessing;

/* Remove the source of the bug */
signed pow(signed, signed) {
  assert(false);
  return -1;
}

/* define hash */
namespace std {
template <>
class hash<std::pair<ll, ll>> {
 public:
  size_t operator()(const std::pair<ll, ll>& x) const {
    return hash<ll>()(1000000000 * x.first + x.second);
  }
};
}  // namespace std

/* input */
template <class T>
std::istream& operator>>(std::istream& is, vector<T>& vec) {
  for (T& x : vec) is >> x;
  return is;
}

//=============================================================================================

constexpr auto nullLambda = [](ll n) {};
template <class Val = bool, Val ignore = false>
class TrieTree {
  Val m_val;
  std::vector<std::unique_ptr<TrieTree>> m_next;
  // static constexpr auto nullLambda = [](ll n) {}; c++17

  auto emplace(const std::vector<int>& vec, Val val, int it) {
    if (it == vec.size()) {
      m_val = val;
      return;
    }
    if (!m_next[vec[it]]) {
      m_next[vec[it]] = std::make_unique<TrieTree>();
    }
    m_next[vec[it]]->emplace(vec, val, it + 1);
  }

  template <class Lambda>
  auto find(const std::vector<int>& vec, int it, const Lambda& lambda) const {
    if (m_val != ignore) {
      lambda(m_val);
    }
    if (it == vec.size()) {
      return m_val;
    }
    if (!m_next[vec[it]]) {
      return ignore;
    }
    return m_next[vec[it]]->find(vec, it + 1, lambda);
  }

 public:
  TrieTree() : TrieTree(ignore) {}
  TrieTree(Val val)
      : m_val(val), m_next(std::vector<std::unique_ptr<TrieTree>>(26)) {}

  auto emplace(const std::vector<int>& vec, Val val) {
    return emplace(vec, val, 0);
  }

  template <class Lambda = decltype(nullLambda)>
  auto find(const std::vector<int>& vec, const Lambda& lambda = nullLambda) {
    return find(vec, 0, lambda);
  }
};

signed main() {
  string str;
  cin >> str;
  ll n;
  cin >> n;

  auto tree = TrieTree<bool, false>();
  REP(_, n) {
    string s;
    cin >> s;
    VL v;
    v.reserve(s.size());
    REPC(c, s) { v.emplace_back(c - 'A'); }
    tree.emplace(v, true);
  }

  ll sz = str.size();
  ll ans = 0;
  REP(i, sz) {
    VL v;
    v.reserve(10);
    FOR(j, i, std::min(i + 10, sz)) { v.emplace_back(str[j] - 'A'); }
    auto update = [&](bool b) { ++ans; };
    tree.find(v, update);
  }
  cout << ans << endl;
}