結果

問題 No.430 文字列検索
コンテスト
ユーザー kopricky
提出日時 2018-07-16 07:31:33
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 28 ms / 2,000 ms
コード長 3,294 bytes
コンパイル時間 1,733 ms
コンパイル使用メモリ 187,824 KB
実行使用メモリ 14,080 KB
最終ジャッジ日時 2024-11-10 00:19:28
合計ジャッジ時間 2,476 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 14
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define ll long long
#define INF 1000000005
#define MOD 1000000007
#define EPS 1e-10
#define rep(i,n) for(int i=0;i<(int)(n);++i)
#define rrep(i,n) for(int i=(int)(n)-1;i>=0;--i)
#define srep(i,s,t) for(int i=(int)(s);i<(int)(t);++i)
#define each(a,b) for(auto& (a): (b))
#define all(v) (v).begin(),(v).end()
#define len(v) (int)(v).size()
#define zip(v) sort(all(v)),v.erase(unique(all(v)),v.end())
#define cmx(x,y) x=max(x,y)
#define cmn(x,y) x=min(x,y)
#define fi first
#define se second
#define pb push_back
#define show(x) cout<<#x<<" = "<<(x)<<endl
#define spair(p) cout<<#p<<": "<<p.fi<<" "<<p.se<<endl
#define sar(a,n) cout<<#a<<":";rep(pachico,n)cout<<" "<<a[pachico];cout<<endl
#define svec(v) cout<<#v<<":";rep(pachico,v.size())cout<<" "<<v[pachico];cout<<endl
#define svecp(v) cout<<#v<<":";each(pachico,v)cout<<" {"<<pachico.first<<":"<<pachico.second<<"}";cout<<endl
#define sset(s) cout<<#s<<":";each(pachico,s)cout<<" "<<pachico;cout<<endl
#define smap(m) cout<<#m<<":";each(pachico,m)cout<<" {"<<pachico.first<<":"<<pachico.second<<"}";cout<<endl

using namespace std;

typedef pair<int,int> P;
typedef pair<ll,ll> pll;
typedef vector<int> vi;
typedef vector<vi> vvi;
typedef vector<ll> vl;
typedef vector<vl> vvl;
typedef vector<double> vd;
typedef vector<P> vp;
typedef vector<string> vs;

const int MAX_N = 100005;

class AhoCorasick
{
private:
    using AC = AhoCorasick*;
    vector<int> ac;
    map<int, AC> next;
    AC fail, dnx;
    void add(const string& s, int pos, int id){
        if(pos == (int)s.size()){
            ac.push_back(id);
            return;
        }
        if(!next[s[pos]]) next[s[pos]] = new AhoCorasick();
        next[s[pos]]->add(s,pos+1,id);
    }
    AC count(vector<int>& res){
        if(!fail) return this;
        for(int id : ac) res[id]++;
        if(dnx) dnx->count(res);
        else dnx = fail->count(res);
        return ac.size() ? this : dnx;
    }
public:
    AhoCorasick() : fail(nullptr), dnx(nullptr){}
    AC build(vector<string>& pattern){
        int sz = (int)pattern.size();
        AC t = new AhoCorasick();
        rep(i,sz) t->add(pattern[i],0,i);
        queue<AC> que;
        que.push(t);
        t->fail = nullptr;
        while(!que.empty()){
            AC nw = que.front();
            que.pop();
            for(auto p : nw->next){
                int i = p.first;
                AC fail = nw->fail;
                while(fail && !fail->next.count(i)){
                    fail = fail->fail;
                }
                nw->next[i]->fail = (!fail ? t : fail->next[i]);
                que.push(nw->next[i]);
            }
        }
        return t;
    }
    void match(vector<int>& res, const string& s, int pos){
        if(pos == (int)s.size()) return;
        if(next[s[pos]]){
            next[s[pos]]->count(res);
            next[s[pos]]->match(res,s,pos+1);
        }else{
            if(!fail) match(res,s,pos+1);
            else fail->match(res,s,pos);
        }
    }
};

int main()
{
    string s;
    cin >> s;
    int m;
    cin >> m;
    vector<string> t(m);
    rep(i,m){
        cin >> t[i];
    }
    AhoCorasick ac;
    auto res = ac.build(t);
    vector<int> ans(m,0);
    res->match(ans,s,0);
    cout << accumulate(all(ans),0) << "\n";
    return 0;
}
0