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

提出ソース
結果

問題	No.430 文字列検索
ユーザー	naskya
提出日時	2021-03-26 23:11:49
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	953 ms / 2,000 ms
コード長	11,858 bytes
コンパイル時間	5,818 ms
コンパイル使用メモリ	352,644 KB
実行使用メモリ	21,376 KB
最終ジャッジ日時	2024-11-10 00:54:35
合計ジャッジ時間	15,447 ms
ジャッジサーバーID （参考情報）	judge4 / judge5
このコードへのチャレンジ
（要ログイン）
テストケース

テストケース表示
入力	結果	実行時間実行使用メモリ
testcase_00	AC	9 ms 11,136 KB
testcase_01	AC	897 ms 21,248 KB
testcase_02	AC	891 ms 21,248 KB
testcase_03	AC	914 ms 21,376 KB
testcase_04	AC	11 ms 11,136 KB
testcase_05	AC	10 ms 11,136 KB
testcase_06	AC	10 ms 11,136 KB
testcase_07	AC	10 ms 11,136 KB
testcase_08	AC	29 ms 21,248 KB
testcase_09	AC	10 ms 11,136 KB
testcase_10	AC	13 ms 12,032 KB
testcase_11	AC	953 ms 21,248 KB
testcase_12	AC	900 ms 21,248 KB
testcase_13	AC	949 ms 21,376 KB
testcase_14	AC	909 ms 21,376 KB
testcase_15	AC	899 ms 21,376 KB
testcase_16	AC	950 ms 21,248 KB
testcase_17	AC	936 ms 21,248 KB
権限があれば一括ダウンロードができます
ソースコード

raw source code
#pragma region template
// clang-format off

#if (defined __INTELLISENSE__) && (!defined PROPER)
#define NDEBUG
namespace std {
#endif

#ifdef LOCAL_NDEBUG
#include <headers.hpp>
#endif

#ifdef LOCAL_DEBUG
#include <debugger.hpp>
#endif

#include <cassert>
#include <cctype>
#include <cmath>
#include <cstddef>
#include <cstdint>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <bitset>
#include <complex>
#include <deque>
#include <functional>
#include <iostream>
#include <iomanip>
#include <limits>
#include <list>
#include <map>
#include <numeric>
#include <queue>
#include <regex>
#include <set>
#include <sstream>
#include <stack>
#include <string>
#include <tuple>
#include <typeinfo>
#include <type_traits>
#include <utility>
#include <vector>

#if (defined __INTELLISENSE__) && (!defined PROPER)
  using namespace std;
}
#endif

#ifdef DEBUGGER_HPP_INCLUDED
#define FILE_LINE "[Debug] ./" + std::string(__FILE__) + ":" + std::to_string(__LINE__) + " in " + std::string(__func__)
#define see(...) debugger::multi_print(#__VA_ARGS__, __VA_ARGS__)
#define seetype(arg) debugger::multi_print(#arg, debugger::type_name<decltype(arg)>)
#define here(...) do {debugger::output << FILE_LINE << ": \e[32mReached\e[39m\n"; see(__VA_ARGS__); debugger::output << (std::string(#__VA_ARGS__).empty() ? "" : "\n");} while(0)
#define com(msg) debugger::output << FILE_LINE << ":\n    \e[36mComment:\e[39m " << msg << "\n"
#define local(x) do {x} while(0)
#define alter(x, y) x
#else
#define see(...) (static_cast<void>(0))
#define seetype(arg) (static_cast<void>(0))
#define here(...) (static_cast<void>(0))
#define com(msg) (static_cast<void>(0))
#define local(x) (static_cast<void>(0))
#define alter(x, y) y
#endif

#if (defined DEBUGGER_HPP_INCLUDED) && (!defined NOWARN)
#define warn(msg) debugger::output << FILE_LINE << ":\n    \e[33mWarning:\e[39m " << msg << "\n"
#else
#define warn(msg) (static_cast<void>(0))
#endif

#if (defined LOCAL_DEBUG) || (defined LOCAL_NDEBUG) || (defined __INTELLISENSE__)
#define NOEXCEPT
#define M_assert(expr) assert(expr)
#define O_assert(expr) assert(expr)
#else
#define NOEXCEPT noexcept
#define M_assert(expr) do {if(__builtin_expect(!(expr), 0)) {auto p = static_cast<std::int64_t*>(std::malloc(1 << 30)); for (int i = 0; i < (1 << 27); p[i] = 1, i += (1 << 9)); std::cerr << (*p);}} while(0)
#define O_assert(expr) do {if(__builtin_expect(!(expr), 0)) {const auto X = std::string(1000, '-'); for(int i = 0; i < (1 << 18); i++) std::cout << X;}} while(0)
#endif

#define rep(i, n) for (std::make_signed_t<std::decay_t<decltype(n)>> i = 0; i < (n); ++i)
#define rng(i, f, t, s) for (std::make_signed_t<std::decay_t<std::common_type_t<decltype(f), decltype(t)>>> i = (f); ((s) > 0) ? (i < (t)) : (i > (t)); i += (s))
#define erng(i, f, t, s) for (std::make_signed_t<std::decay_t<std::common_type_t<decltype(f), decltype(t)>>> i = (f); ((s) > 0) ? (i <= (t)) : (i >= (t)); i += (s))

[[maybe_unused]] constexpr int         INF   = 1000000005;
[[maybe_unused]] constexpr long long   LINF  = 1000000000000000005LL;
[[maybe_unused]] constexpr double      EPS   = 1e-9;
[[maybe_unused]] constexpr long double LEPS  = 1e-14L;
[[maybe_unused]] constexpr int         dy[9] = {1, 0, -1, 0, 1, 1, -1, -1, 0};
[[maybe_unused]] constexpr int         dx[9] = {0, 1, 0, -1, -1, 1, 1, -1, 0};

template <class S, class T, class... U, class V = std::common_type_t<S, T, U...>> constexpr V Min(const S a, const T b, const U... c) {
  if constexpr (sizeof...(U)) return std::min((V) a, (V) Min(b, c...)); else return std::min((V) a, (V) b);
}
template <class S, class T, class... U, class V = std::common_type_t<S, T, U...>> constexpr V Max(const S a, const T b, const U... c) {
  if constexpr (sizeof...(U)) return std::max((V) a, (V) Max(b, c...)); else return std::max((V) a, (V) b);
}

// clang-format on
#pragma endregion

#define ROLLING_HASH_MAX_LENGTH 500050
#define NUMBER_OF_BASES 2

#pragma region lib_rolling_hash

namespace lib {
  namespace internal {
    template <std::size_t N> using hash_type = std::conditional_t<(30 < N), std::uint64_t, std::uint32_t>;

    template <std::size_t N> constexpr hash_type<N> mask = (static_cast<hash_type<N>>(1) << N) - 1;
    template <std::size_t N> constexpr hash_type<N> mul_mod_1(const hash_type<N> lhs, const hash_type<N> rhs) {
      const hash_type<N> lhs_u = lhs >> N, lhs_d = lhs & mask<N>;
      const hash_type<N> rhs_u = rhs >> N, rhs_d = rhs & mask<N>;
      const hash_type<N> mid = lhs_d * rhs_u + lhs_u * rhs_d;
      return ((lhs_u * rhs_u) << 1) + lhs_d * rhs_d + ((mid & mask<N - 1>) << N) + (mid >> (N - 1));
    }
    template <std::size_t N, class Tp1, class Tp2> constexpr Tp1 mul_mod_2(const Tp1 lhs, const Tp2 rhs) {
      const Tp2 rhs_d = rhs & mask<N>;
      const Tp1 lhs_u = lhs >> N, mid = lhs_u * rhs_d;
      return (lhs & mask<N>) *rhs_d + ((mid & mask<N - 1>) << N) + (mid >> (N - 1));
    }
    template <std::size_t N> constexpr hash_type<N> clamp(const hash_type<N> x) {
      if (const hash_type<N> res = (x >> N) + (x & mask<N>); res < mask<N>)
        return res;
      else
        return res - mask<N>;
    }
  }  // namespace internal

  namespace internal::rolling_hash_61 {
    using HashType = hash_type<61>;

    constexpr int MAX_N              = ROLLING_HASH_MAX_LENGTH;
    constexpr int BASES_N            = NUMBER_OF_BASES;
    constexpr std::uint32_t bases[5] = {27644437, 479001599, 998244353, 999999937, 1073676287};
    constexpr HashType padding       = mask<61> * mask<std::numeric_limits<HashType>::digits - 61>;

    struct base_pow_constexpr {
      HashType val[BASES_N][MAX_N + 1];

      constexpr base_pow_constexpr() : val() {
        static_assert(BASES_N < 5);

        for (int i = 0; i < BASES_N; ++i)
          val[i][0] = 1;

        for (int i = 0; i < BASES_N; ++i)
          for (int j = 1; j <= MAX_N; ++j)
            val[i][j] = clamp<61>(mul_mod_2<31>(val[i][j - 1], bases[i]));
      }
    };

#if (ROLLING_HASH_MAX_LENGTH * NUMBER_OF_BASES > 250000)
    // if you want ROLLING_HASH_MAX_LENGTH * NUMBER_OF_BASES to be greater than
    // constexpr-loop-limit (typically 262144 = 2^18 in GCC), you should use
    // `const` instead of `constexpr` (because you cannot specify -fconstexpr-loop-limit=*** on online judge)
    const base_pow_constexpr base_to_the_n;
#else
    constexpr base_pow_constexpr base_to_the_n;
#endif

    std::vector<std::array<HashType, BASES_N>> build_hash(const std::string& s) {
      const int N = static_cast<int>(s.size());
      O_assert(N <= MAX_N);

      std::vector<std::array<HashType, BASES_N>> res(N + 1);
      std::fill(res.front().begin(), res.front().end(), 0);

      for (int i = 1; i <= N; ++i)
        for (int j = 0; j < BASES_N; ++j)
          res[i][j] = clamp<61>(mul_mod_2<31>(res[i - 1][j], bases[j]) + static_cast<HashType>(s[i - 1]));

      return res;
    }

    std::array<HashType, BASES_N> substr_hash(const std::vector<std::array<HashType, BASES_N>>& hash_vec,
                                              const int L_index, int cnt = std::numeric_limits<int>::max()) {
      const int N = static_cast<int>(hash_vec.size());
      O_assert(0 <= L_index && L_index < N);
      local(if (N - L_index < cnt && cnt != std::numeric_limits<int>::max()) warn("substr was shortened (" << cnt << " -> " << (N - L_index) << ")"););
      cnt = std::min(cnt, N - L_index);

      std::array<HashType, BASES_N> res;
      for (int i = 0; i < BASES_N; ++i)
        res[i] = clamp<61>(hash_vec[L_index + cnt][i] + padding - mul_mod_1<31>(hash_vec[L_index][i], base_to_the_n.val[i][cnt]));

      return res;
    }
  }  // namespace internal::rolling_hash_61

#ifdef LOCAL_DEBUG
  namespace internal {
    class string_hash {
      public:
      using type = std::array<internal::rolling_hash_61::HashType, internal::rolling_hash_61::BASES_N>;

      string_hash(void) noexcept {}
      string_hash(const type h, const std::string& s) : v(h), source(s) {}

      bool operator==(const string_hash rhs) const {
        const bool same_hash = (v == rhs.v), same_string = (source == rhs.source);
        assert(same_hash == same_string);
        return same_string;
      }
      bool operator!=(const string_hash rhs) const {
        const bool same_hash = (v == rhs.v), same_string = (source == rhs.source);
        assert(same_hash == same_string);
        return !same_string;
      }

      type hash_vals(void) const noexcept {
        return v;
      }

      private:
      type v;
      std::string source;
    };
  }  // namespace internal

  class rolling_hash {
    public:
    using type = std::array<internal::rolling_hash_61::HashType, internal::rolling_hash_61::BASES_N>;

    rolling_hash(void) noexcept {}
    rolling_hash(const std::string& s) : source(s), hash(internal::rolling_hash_61::build_hash(s)) {}

    // get hash of the whole string
    internal::string_hash whole_string(void) const {
      return internal::string_hash(hash.back(), source);
    }

    // get hash of the whole string
    internal::string_hash substr(void) const {
      return internal::string_hash(hash.back(), source);
    }

    // get hash of the substring of length len, starting at L_index
    internal::string_hash substr(const int L_index, int len = std::numeric_limits<int>::max()) const {
      return internal::string_hash(internal::rolling_hash_61::substr_hash(hash, L_index, len), source.substr(L_index, len));
    }

    private:
    std::vector<type> hash;
    std::string source;
  };
#else
  namespace internal {
    class string_hash {
      public:
      using type = std::array<internal::rolling_hash_61::HashType, internal::rolling_hash_61::BASES_N>;

      string_hash(void) noexcept {}
      string_hash(const type h) noexcept : v(h) {}

      bool operator==(const string_hash rhs) const noexcept {
        return v == rhs.v;
      }
      bool operator!=(const string_hash rhs) const noexcept {
        return v != rhs.v;
      }

      type hash_vals(void) const noexcept {
        return v;
      }

      private:
      type v;
    };
  }  // namespace internal

  class rolling_hash {
    public:
    using type = std::array<internal::rolling_hash_61::HashType, internal::rolling_hash_61::BASES_N>;

    rolling_hash(void) noexcept {}
    rolling_hash(const std::string& s) : hash(internal::rolling_hash_61::build_hash(s)) {}

    // get hash of the whole string
    internal::string_hash whole_string(void) const {
      return internal::string_hash(hash.back());
    }

    // get hash of the whole string
    internal::string_hash substr(void) const {
      return internal::string_hash(hash.back());
    }

    // get hash of the substring of length len, starting at L_index
    internal::string_hash substr(const int L_index, int len = std::numeric_limits<int>::max()) const {
      return internal::string_hash(internal::rolling_hash_61::substr_hash(hash, L_index, len));
    }

    private:
    std::vector<type> hash;
  };
#endif
}  // namespace lib

#pragma endregion

void solve() {
  std::string S;
  int M;
  std::cin >> S >> M;

  const lib::rolling_hash S_hash(S);
  const int S_len = static_cast<int>(S.size());

  int ans = 0;
  std::vector precalc_hash(11, std::vector<lib::rolling_hash::type>(S_len));

  erng(len, 1, 10, 1) {
    rep(start, S_len - len + 1) {
      precalc_hash[len][start] = S_hash.substr(start, len).hash_vals();
    }
  }

  while (M--) {
    std::string C;
    std::cin >> C;
    const auto C_whole_hash = lib::rolling_hash(C).whole_string().hash_vals();
    const int C_len         = static_cast<int>(C.size());

    rep(start, S_len - C_len + 1) {
      if (precalc_hash[C_len][start] == C_whole_hash)
        ans++;
    }
  }

  std::cout << ans << "\n";
}

int main() {
  std::ios_base::sync_with_stdio(false);
  std::cin.tie(nullptr);
  std::cout.tie(nullptr);
  solve();
}
yukicoder

結果

テストケース

ソースコード
