結果

問題 No.430 文字列検索
コンテスト
ユーザー y
提出日時 2021-05-25 15:39:18
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,551 ms / 2,000 ms
コード長 4,136 bytes
コンパイル時間 1,440 ms
コンパイル使用メモリ 173,416 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-11-10 00:56:44
合計ジャッジ時間 12,679 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 14
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
// #define LOCAL // 提出時はコメントアウト
#define DEBUG_

typedef long long ll;
const double EPS = 1e-9;
const ll INF = ((1LL<<62)-(1LL<<31));
typedef vector<ll> vecl;
typedef pair<ll, ll> pairl;
template<typename T> using uset = unordered_set<T>;
template<typename T, typename U> using mapv = map<T,vector<U>>;
template<typename T, typename U> using umap = unordered_map<T,U>;

#define ALL(v) v.begin(), v.end()
#define REP(i, x, n) for(int i = x; i < n; i++)
#define rep(i, n) REP(i, 0, n)
#define sz(x) (ll)x.size()
ll llceil(ll a,ll b) { return (a+b-1)/b; }
template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return true; } return false; }
template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return true; } return false; }
template<class T> vector<vector<T>> genarr(ll n, ll m, T init) { return vector<vector<T>>(n,vector<T>(m,init)); }

///// DEBUG
#define DUMPOUT cerr
#define repi(itr, ds) for (auto itr = ds.begin(); itr != ds.end(); itr++)
template<typename T>istream&operator>>(istream&is,vector<T>&vec){for(T&x:vec)is>>x;return is;}
template<typename T,typename U>ostream&operator<<(ostream&os,pair<T,U>&pair_var){os<<"("<<pair_var.first<<", "<<pair_var.second<<")";return os;}
template<typename T>ostream&operator<<(ostream&os,const vector<T>&vec){os<<"{";for(int i=0;i<vec.size();i++){os<<vec[i]<<(i+1==vec.size()?"":", ");}
os<<"}";return os;}
template<typename T,typename U>ostream&operator<<(ostream&os,map<T,U>&map_var){os<<"{";repi(itr,map_var){os<<*itr;itr++;if(itr!=map_var.end())os<<", ";itr--;}
os<<"}";return os;}
template<typename T>ostream&operator<<(ostream&os,set<T>&set_var){os<<"{";repi(itr,set_var){os<<*itr;itr++;if(itr!=set_var.end())os<<", ";itr--;}
os<<"}";return os;}
void dump_func(){DUMPOUT<<endl;}
template<class Head,class...Tail>void dump_func(Head&&head,Tail&&...tail){DUMPOUT<<head;if(sizeof...(Tail)>0){DUMPOUT<<", ";}
dump_func(std::move(tail)...);}
#ifndef LOCAL
#undef DEBUG_
#endif
#ifdef DEBUG_
#define DEB
#define dump(...)                                                          \
DUMPOUT << "  " << string(#__VA_ARGS__) << ": "                            \
        << "[" << to_string(__LINE__) << ":" << __FUNCTION__ << "]"        \
        << endl                                                            \
        << "    ",                                                         \
    dump_func(__VA_ARGS__)
#else
#define DEB if (false)
#define dump(...)
#endif

//////////

#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")

struct KMP {
    string P;
    vecl table; // table[i]: iでfailした時, どこから検索し直すか
    KMP(string pattern) : P(pattern) {
        table.assign(sz(P)+1,0LL); 
        table[0] = -1;
        for (int i = 0, p = -1; i < sz(P); i++) {
            while (p >= 0 && P[i] != P[p]) p = table[p]; // P[i]と重複している先頭部分P[0:p]を探す
            table[i+1] = ++p; // P[0:p]が一致しているので, i+1でfailした場合, 検索し直す先頭はP[p+1]から
        }
    }

    ll count(string S) { // O(N)
        ll res = 0;
        for (int i = 0, p = 0; i < sz(S); i++) {
            while (p >= 0 && S[i] != P[p]) p = table[p];
            if (p == sz(P) - 1) res++;
            p++;
        }
        return res;
    }

    ll find(string S) { // マッチしたSの先頭indexを返す, O(N)
        for (int i = 0, p = 0; i < sz(S); i++) {
            while (p >= 0 && S[i] != P[p]) p = table[p];
            if (p == sz(P) - 1) return i - sz(P) + 1;
            p++;
        }
        return -1;
    }
};

int solve(ostringstream &cout) {
    #ifdef LOCAL
    ifstream in("../../Atcoder/input.txt");
    cin.rdbuf(in.rdbuf());
    #endif

    string S;
    cin>>S;

    ll M;
    cin>>M;
    
    ll ans = 0;
    rep(q,M) {
        string C;
        cin>>C; 
        auto kmp = KMP(C);
        ans += kmp.count(S);
    }

    cout << ans << endl;

    return 0;
}

int main() {
    ostringstream oss;
    int res = solve(oss);

    cout << oss.str();
    return res;
}
0