結果

問題 No.430 文字列検索
コンテスト
ユーザー kopricky
提出日時 2018-09-07 22:19:49
言語 C++14
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 37 ms / 2,000 ms
コード長 3,714 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,128 ms
コンパイル使用メモリ 190,760 KB
実行使用メモリ 15,616 KB
最終ジャッジ日時 2024-11-10 00:19:46
合計ジャッジ時間 3,027 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 14
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#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<pair<int, int> > ac;
    map<int, AC> next;
    AC t, fail, dnx;
    void add(const string& s, int pos, int id){
        if(pos == (int)s.size()){
            ac.emplace_back(id, pos);
            return;
        }
        if(!next[s[pos]]) next[s[pos]] = new AhoCorasick();
        next[s[pos]]->add(s,pos+1,id);
    }
    AC count(vector<vector<int> >& res, int pos){
        if(!fail) return this;
        for(auto val : ac) res[val.first].push_back(pos-val.second+1);
        if(dnx) dnx->count(res,pos);
        else dnx = fail->count(res,pos);
        return ac.size() ? this : dnx;
    }
    void match_(vector<vector<int> >& res, const string& s, int pos=0){
        if(pos == (int)s.size()) return;
        if(next[s[pos]]){
            next[s[pos]]->count(res, pos);
            next[s[pos]]->match_(res,s,pos+1);
        }else{
            if(!fail) match_(res,s,pos+1);
            else fail->match_(res,s,pos);
        }
    }
public:
    AhoCorasick() : fail(nullptr), dnx(nullptr){}
    //パターンを入力として木を構築
    AhoCorasick(vector<string>& pattern): AhoCorasick(){
        int sz = (int)pattern.size();
        t = new AhoCorasick();
        for(int i = 0; i < sz; i++){
            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]);
            }
        }
    }
    //sのインデックスpos(デフォルト0)以降に含まれる各パターンの開始位置をresで返す
    void match(vector<vector<int> >& res, const string& s, int pos=0){
        t->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(t);
    vvi ans(m);
    ac.match(ans,s);
    int sm = 0;
    rep(i, m){
        sm += len(ans[i]);
    }
    cout << sm << "\n";
    return 0;
}
0