#467487 (C++11) No.430 文字列検索

提出ソース

結果

問題	No.430 文字列検索
ユーザー	hamray
提出日時	2020-04-20 17:36:52
言語	C++11 (gcc 11.4.0)
結果	TLE (最新) AC (最初)
実行時間	-
コード長	9,183 bytes
コンパイル時間	1,806 ms
コンパイル使用メモリ	174,492 KB
実行使用メモリ	6,824 KB
最終ジャッジ日時	2024-11-10 00:43:18
合計ジャッジ時間	30,234 ms
ジャッジサーバーID （参考情報）	judge2 / judge4

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	TLE	-
testcase_02	TLE	-
testcase_03	TLE	-
testcase_04	AC	2 ms 5,248 KB
testcase_05	AC	2 ms 5,248 KB
testcase_06	AC	2 ms 5,248 KB
testcase_07	AC	2 ms 5,248 KB
testcase_08	AC	12 ms 5,248 KB
testcase_09	AC	2 ms 5,248 KB
testcase_10	AC	8 ms 5,248 KB
testcase_11	TLE	-
testcase_12	TLE	-
testcase_13	TLE	-
testcase_14	TLE	-
testcase_15	TLE	-
testcase_16	TLE	-
testcase_17	TLE	-

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>
//typedef
//-------------------------#include <bits/stdc++.h>

const double pi = 3.141592653589793238462643383279;

using namespace std;

//conversion
//------------------------------------------
inline int toInt(string s)
{
  int v;
  istringstream sin(s);
  sin >> v;
  return v;
}
template <class T>
inline string toString(T x)
{
  ostringstream sout;
  sout << x;
  return sout.str();
}
inline int readInt()
{
  int x;
  scanf("%d", &x);
  return x;
}

//typedef
//------------------------------------------
typedef vector<int> VI;
typedef vector<VI> VVI;
typedef vector<string> VS;
typedef pair<int, int> PII;
typedef pair<long long, long long> PLL;
typedef pair<int, PII> TIII;
typedef long long LL;
typedef unsigned long long ULL;
typedef vector<LL> VLL;
typedef vector<VLL> VVLL;

//container util

//------------------------------------------
#define ALL(a) (a).begin(), (a).end()
#define RALL(a) (a).rbegin(), (a).rend()
#define PB push_back
#define MP make_pair
#define SZ(a) int((a).size())
#define SQ(a) ((a) * (a))
#define EACH(i, c) for (typeof((c).begin()) i = (c).begin(); i != (c).end(); ++i)
#define EXIST(s, e) ((s).find(e) != (s).end())
#define SORT(c) sort((c).begin(), (c).end())

//repetition
//------------------------------------------
#define FOR(i, s, n) for (int i = s; i < (int)n; ++i)
#define REP(i, n) FOR(i, 0, n)
#define MOD 1000000007

#define rep(i, a, b) for (int i = a; i < (b); ++i)
#define trav(a, x) for (auto &a : x)
#define all(x) x.begin(), x.end()
#define sz(x) (int)(x).size()

typedef long long ll;
typedef pair<int, int> P;
typedef pair<int, int> pii;
typedef vector<int> vi;
const double EPS = 1E-8;

#define chmin(x, y) x = min(x, y)
#define chmax(x, y) x = max(x, y)

class UnionFind
{
public:
  vector<int> par;
  vector<int> siz;

  UnionFind(int sz_) : par(sz_), siz(sz_, 1)
  {
    for (ll i = 0; i < sz_; ++i)
      par[i] = i;
  }
  void init(int sz_)
  {
    par.resize(sz_);
    siz.assign(sz_, 1LL);
    for (ll i = 0; i < sz_; ++i)
      par[i] = i;
  }

  int root(int x)
  {
    if (par[x] == x)
    {
      return x;
    }
    else
    {
      return par[x] = root(par[x]);
    }
  }

  bool merge(int x, int y)
  {
    x = root(x);
    y = root(y);
    if (x == y)
      return false;
    if (siz[x] < siz[y])
      swap(x, y);
    siz[x] += siz[y];
    par[y] = x;
    return true;
  }

  bool issame(int x, int y)
  {
    return root(x) == root(y);
  }

  int size(int x)
  {
    return siz[root(x)];
  }
};

ll modPow(ll x, ll n, ll mod = MOD)
{
  if (n == 0)
    return 1;
  ll res = 1;
  while (n)
  {
    if (n & 1)
      res = (res * x) % mod;

    res %= mod;
    x = x * x % mod;
    n >>= 1;
  }
  return res;
}

#define SIEVE_SIZE 5000000 + 10
bool sieve[SIEVE_SIZE];
void makeSieve()
{
  for (int i = 0; i < SIEVE_SIZE; ++i)
    sieve[i] = true;
  sieve[0] = sieve[1] = false;
  for (int i = 2; i * i < SIEVE_SIZE; ++i)
    if (sieve[i])
      for (int j = 2; i * j < SIEVE_SIZE; ++j)
        sieve[i * j] = false;
}

bool isprime(ll n)
{
  if (n == 0 || n == 1)
    return false;
  for (ll i = 2; i * i <= n; ++i)
    if (n % i == 0)
      return false;
  return true;
}

const int MAX = 3000010;
long long fac[MAX], finv[MAX], inv[MAX];

const ll mod2 = 998244353;
const ll mod3 = 1777777777;
// テーブルを作る前処理
void COMinit()
{
  fac[0] = fac[1] = 1;
  finv[0] = finv[1] = 1;
  inv[1] = 1;
  for (ll i = 2; i < MAX; i++)
  {
    fac[i] = fac[i - 1] * i % MOD;
    inv[i] = MOD - inv[MOD % i] * (MOD / i) % MOD;
    finv[i] = finv[i - 1] * inv[i] % MOD;
    //cout << fac[i] << " " << inv[i] << " " << finv[i] << endl;
  }
}

// 二項係数計算
long long COM(int n, int k)
{
  if (n < k)
    return 0;
  if (n < 0 || k < 0)
    return 0;
  return fac[n] * (finv[k] * finv[n - k] % MOD) % MOD;
}

long long extGCD(long long a, long long b, long long &x, long long &y)
{
  if (b == 0)
  {
    x = 1;
    y = 0;
    return a;
  }
  long long d = extGCD(b, a % b, y, x);
  y -= a / b * x;
  return d;
}
// 負の数にも対応した mod (a = -11 とかでも OK)
inline long long mod(long long a, long long m)
{
  return (a % m + m) % m;
}

// 逆元計算 (ここでは a と m が互いに素であることが必要)
long long modinv(long long a, long long m)
{
  long long x, y;
  extGCD(a, m, x, y);
  return mod(x, m); // 気持ち的には x % m だが、x が負かもしれないので
}
ll GCD(ll a, ll b)
{

  if (b == 0)
    return a;
  return GCD(b, a % b);
}

typedef vector<ll> vec;
typedef vector<vec> mat;

mat mul(mat &A, mat &B)
{
  mat C(A.size(), vec((int)B[0].size()));
  for (int i = 0; i < A.size(); ++i)
  {
    for (int k = 0; k < B.size(); ++k)
    {
      for (int j = 0; j < B[0].size(); ++j)
      {
        C[i][j] = (C[i][j] + A[i][k] % MOD * B[k][j] % MOD) % MOD;
      }
    }
  }
  return C;
}
mat matPow(mat A, ll n)
{
  mat B(A.size(), vec((int)A.size()));

  for (int i = 0; i < A.size(); ++i)
  {
    B[i][i] = 1;
  }

  while (n > 0)
  {
    if (n & 1)
      B = mul(B, A);
    A = mul(A, A);
    n >>= 1;
  }
  return B;
}

template <class TYPE>
struct Fenwick
{
  int N;
  vector<ll> bit;

  Fenwick() : N(0) {}
  Fenwick(int n) : N(n), bit(n + 1, 0) {}

  void add(int i, ll x)
  {
    //0-indexedを避けるために加算
    while (i <= N)
    {
      bit[i] += x;
      i += (i & -i);
    }
  }

  ll sum(int i)
  {
    ll res = 0;
    while (i > 0)
    {
      res += bit[i];
      i -= (i & -i);
    }
    return res;
  }

  ll range(int i, int j)
  {
    return sum(j) - sum(i);
  }

  ll get(int i)
  {
    return sum(i) - sum(i - 1);
  }

  ll getKth(ll k)
  {
    ++k;
    int res = 0;
    int n = 1;
    while (n < bit.size())
      n *= 2;

    for (int i = n / 2; i > 0; i /= 2)
    {
      if (res + i < bit.size() && bit[res + i] < k)
      {
        k = k - bit[res + i];
        res = res + i;
      }
    }
    return res + 1;
  }

  // bitは1-indexedで実装しているのでバグった時は初めに確認すること
};
// 各iに対して文字列S[0,i-1]の接頭辞と接尾辞が最大何文字一致しているか
struct RollingHash
{

  using long_type = unsigned;
  using hash_type = tuple<long_type, long_type, long_type>;

private:
  int s_len;

  long_type mod1;
  long_type mod2;
  long_type mod3;

  long_type base1;
  long_type base2;
  long_type base3;

  vector<long_type> hash1;
  vector<long_type> hash2;
  vector<long_type> hash3;

  vector<long_type> pow1;
  vector<long_type> pow2;
  vector<long_type> pow3;

public:
  RollingHash() : base1(1009), base2(1007), base3(2339),
                  mod1(1000000007), mod2(1000000009), mod3(1000000087)
  {
  }

  RollingHash(const string s) : RollingHash()
  {
    init(s);
  }

  void init(const string s)
  {

    const int n = s.size();
    s_len = n;

    hash1.assign(n + 1, 0);
    hash2.assign(n + 1, 0);
    hash3.assign(n + 1, 0);
    pow1.assign(n + 1, 1);
    pow2.assign(n + 1, 1);
    pow3.assign(n + 1, 1);

    for (int i = 0; i < n; i++)
    {
      hash1[i + 1] = ((hash1[i] + s[i]) * base1);
      hash2[i + 1] = ((hash2[i] + s[i]) * base2);
      hash3[i + 1] = ((hash3[i] + s[i]) * base3);
      pow1[i + 1] = (pow1[i] * base1);
      pow2[i + 1] = (pow2[i] * base2);
      pow3[i + 1] = (pow3[i] * base3);
    }
  }

  //0-index
  //s[l, r)のハッシュタプルを返す
  hash_type get_hash_tuple(int l, int r)
  {
    long_type h1 = ((hash1[r] - hash1[l] * pow1[r - l] + mod1) % mod1 + mod1) % mod1;
    long_type h2 = ((hash2[r] - hash2[l] * pow2[r - l] + mod2) % mod2 + mod2) % mod2;
    long_type h3 = ((hash3[r] - hash3[l] * pow3[r - l] + mod3) % mod3 + mod3) % mod3;
    return make_tuple(h1, h2, h3);
  }

  //not verify
  //htype_1のハッシュの後ろに長さh2_lenのハッシュを結合する
  hash_type concat(hash_type h_tup1, hash_type h_tup2, int h2_len)
  {
    long_type h1 = (get<0>(h_tup1) * pow1[h2_len] + get<0>(h_tup2)) % mod1;
    long_type h2 = (get<1>(h_tup1) * pow2[h2_len] + get<1>(h_tup2)) % mod2;
    long_type h3 = (get<2>(h_tup1) * pow3[h2_len] + get<2>(h_tup2)) % mod3;
    return make_tuple(h1, h2, h3);
  }

  //先頭i文字のハッシュタプルを返す
  hash_type get_from_head(int len)
  {
    return get_hash_tuple(0, len);
  }

  //末尾i文字のハッシュタプルを返す
  hash_type get_from_tail(int len)
  {
    return get_hash_tuple(s_len - len, s_len);
  }

  //sを含む個数の計算
  int contain(string s)
  {
    int ret = 0;
    int m_len = this->s_len;
    int o_len = s.size();
    RollingHash o_hash(s);
    for (int i = 0; i < m_len - o_len + 1; i++)
    {
      auto h1 = get_hash_tuple(i, i + o_len);
      auto h2 = o_hash.get_hash_tuple(0, o_len);
      if (RollingHash::same(h1, h2))
        ret++;
    }
    return ret;
  }

  static bool same(hash_type h_tup1, hash_type h_tup2)
  {
    return h_tup1 == h_tup2;
  }
};
int main()
{
  cin.tie(0);
  ios::sync_with_stdio(false);
  cout << fixed << setprecision(4);

  string s;
  cin >> s;
  int M;
  cin >> M;

  RollingHash rh(s);
  ll ans = 0;
  for (int i = 0; i < M; i++)
  {
    string t;
    cin >> t;
    ans += rh.contain(t);
  }
  cout << ans << endl;
  return 0;
}

yukicoder

結果

テストケース

ソースコード