結果

問題 No.430 文字列検索
ユーザー latte0119
提出日時 2019-11-02 19:56:39
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 94 ms / 2,000 ms
コード長 2,221 bytes
コンパイル時間 1,439 ms
コンパイル使用メモリ 172,144 KB
実行使用メモリ 70,112 KB
最終ジャッジ日時 2024-11-10 00:36:30
合計ジャッジ時間 2,336 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 14
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include<bits/stdc++.h>
using namespace std;
#define int long long 
#define rep(i,n) for(int i=0;i<(n);i++)
#define pb push_back
#define eb emplace_back
#define all(v) (v).begin(),(v).end()
#define fi first
#define se second
typedef vector<int>vint;
typedef pair<int,int>pint;
typedef vector<pint>vpint;
template<typename A,typename B>inline void chmin(A &a,B b){if(a>b)a=b;}
template<typename A,typename B>inline void chmax(A &a,B b){if(a<b)a=b;}
template<class A,class B>
ostream& operator<<(ostream& ost,const pair<A,B>&p){
    ost<<"{"<<p.first<<","<<p.second<<"}";
    return ost;
}
template<class T>
ostream& operator<<(ostream& ost,const vector<T>&v){
    ost<<"{";
    for(int i=0;i<v.size();i++){
        if(i)ost<<",";
        ost<<v[i];
    }
    ost<<"}";
    return ost;
}
struct AhoCorasick{
    struct Node{
        int nex[256];
        int failink;
        vector<int>suflist;
        Node():failink(0){memset(nex,-1,sizeof(nex));}
    };
    vector<Node>nds;
    int process(int k,int c){
        while(nds[k].nex[c]==-1)k=nds[k].failink;
        return nds[k].nex[c];
    }
    void build(const vector<string>&ps){
        nds.emplace_back();
        for(int i=0;i<ps.size();i++){
            int k=0;
            for(auto c:ps[i]){
                if(nds[k].nex[c]==-1){
                    nds[k].nex[c]=nds.size();
                    nds.emplace_back();
                }
                k=nds[k].nex[c];
            }
            nds[k].suflist.push_back(i);
        }
        queue<int>que;
        for(int i=1;i<256;i++){
            if(nds[0].nex[i]==-1){
                nds[0].nex[i]=0;
            }
            else{
                que.push(nds[0].nex[i]);
            }
        }
        while(que.size()){
            int k=que.front();
            que.pop();
            for(int i=1;i<256;i++){
                if(nds[k].nex[i]==-1)continue;
                int nx=nds[k].nex[i];
                nds[nx].failink=process(nds[k].failink,i);
                vector<int>&curlist=nds[nds[nx].failink].suflist;
                vector<int>&nexlist=nds[nx].suflist;
                int tmp=nexlist.size();
                nexlist.insert(nexlist.end(),curlist.begin(),curlist.end());
                //inplace_merge(nexlist.begin(),nexlist.begin()+tmp,nexlist.end());
                que.push(nx);
            }
        }
    }
};
signed main(){
    string S;cin>>S;
    int M;cin>>M;
    vector<string>ps(M);rep(i,M)cin>>ps[i];
    AhoCorasick ac;
    ac.build(ps);
    int ans=0;
    int k=0;
    for(auto c:S){
        while(ac.nds[k].nex[c]==-1)k=ac.nds[k].failink;
        k=ac.nds[k].nex[c];
        ans+=ac.nds[k].suflist.size();
    }
    cout<<ans<<endl;
    return 0;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0