結果

問題 No.430 文字列検索
ユーザー koprickykopricky
提出日時 2018-07-16 07:31:33
言語 C++14
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 29 ms / 2,000 ms
コード長 3,294 bytes
コンパイル時間 1,843 ms
コンパイル使用メモリ 186,896 KB
実行使用メモリ 13,792 KB
最終ジャッジ日時 2023-08-09 01:08:50
合計ジャッジ時間 3,171 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge12
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,384 KB
testcase_01 AC 29 ms
13,792 KB
testcase_02 AC 10 ms
9,892 KB
testcase_03 AC 10 ms
10,088 KB
testcase_04 AC 1 ms
4,384 KB
testcase_05 AC 1 ms
4,380 KB
testcase_06 AC 1 ms
4,380 KB
testcase_07 AC 2 ms
4,380 KB
testcase_08 AC 7 ms
6,280 KB
testcase_09 AC 1 ms
4,384 KB
testcase_10 AC 3 ms
4,380 KB
testcase_11 AC 18 ms
12,288 KB
testcase_12 AC 20 ms
12,640 KB
testcase_13 AC 20 ms
12,732 KB
testcase_14 AC 15 ms
11,508 KB
testcase_15 AC 13 ms
10,544 KB
testcase_16 AC 12 ms
10,564 KB
testcase_17 AC 11 ms
10,484 KB
権限があれば一括ダウンロードができます

ソースコード

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