結果

問題 No.430 文字列検索
コンテスト
ユーザー MitI_7
提出日時 2018-10-12 21:53:30
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 28 ms / 2,000 ms
コード長 4,236 bytes
コンパイル時間 2,036 ms
コンパイル使用メモリ 191,436 KB
実行使用メモリ 15,216 KB
最終ジャッジ日時 2024-11-10 00:20:45
合計ジャッジ時間 2,460 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 14
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

#define FOR(i,a,b) for(int i= (a); i<((int)b); ++i)
#define RFOR(i,a) for(int i=(a); i >= 0; --i)
#define FOE(i,a) for(auto i : a)
#define ALL(c) (c).begin(), (c).end()
#define RALL(c) (c).rbegin(), (c).rend()
#define DUMP(x)  cerr << #x << " = " << (x) << endl;
#define SUM(x) std::accumulate(ALL(x), 0LL)
#define MIN(v) *std::min_element(v.begin(), v.end())
#define MAX(v) *std::max_element(v.begin(), v.end())
#define EXIST(v,x) (std::find(v.begin(), v.end(), x) != v.end())
#define BIT(n) (1LL<<(n))
#define UNIQUE(v) v.erase(unique(v.begin(), v.end()), v.end());

typedef long long LL;
template<typename T> using V = std::vector<T>;
template<typename T> using VV = std::vector<std::vector<T>>;
template<typename T> using VVV = std::vector<std::vector<std::vector<T>>>;

template<class T> inline T ceil(T a, T b) { return (a + b - 1) / b; }
template<class T> inline void print(T x) { std::cout << x << std::endl; }
template<class T> inline void print_vec(const std::vector<T> &v) { for (int i = 0; i < v.size(); ++i) {  if (i != 0) {std::cout << " ";} std::cout << v[i];} std::cout << "\n"; }
template<class T> inline bool inside(T y, T x, T H, T W) {return 0 <= y and y < H and 0 <= x and x < W; }
template<class T> inline double euclidean_distance(T y1, T x1, T y2, T x2) { return sqrt((x1 - x2) * (x1 - x2) + (y1 - y2) * (y1 - y2)); }
template<class T> inline double manhattan_distance(T y1, T x1, T y2, T x2) { return abs(x1 - x2) + abs(y1 - y2); }

const int INF = 1L << 30;
const double EPS = 1e-9;
const std::string YES = "YES", Yes = "Yes", NO = "NO", No = "No";
const std::vector<int> dy4 = { 0, 1, 0, -1 }, dx4 = { 1, 0, -1, 0 };    // 4近傍（右, 下, 左, 上）
const std::vector<int> dy8 = { 0, -1, 0, 1, 1, -1, -1, 1 }, dx8 = { 1, 0, -1, 0, 1, 1, -1, -1 };

class AhoCorasick {
    struct Node {
        std::unordered_map<int, Node*> next;
        Node* failure;
        std::vector<std::string> output;

        Node() {
            failure = nullptr;
        };

        void insert(int i, const std::string &s) {
            if (i >= s.size()) {
                output.emplace_back(s);
                return;
            }

            if (this->next[s[i]] == nullptr) {
                this->next[s[i]] = new Node();
            }

            this->next[s[i]]->insert(i + 1, s);
        }
    };

public:
    Node root;

    void insert(const std::string &word) {
        root.insert(0, word);
    }

    void build() {
        std::queue<Node*> que;
        que.push(&root);
        while (not que.empty()) {
            auto node = que.front(); que.pop();

            for (auto &p : node->next) {
                auto now = node;
                auto next = p.second;
                auto c = p.first;
                que.push(next);

                while (now != &root and now->failure->next[c] == nullptr) {
                    now = now->failure;
                }

                if (now == &root) {
                    next->failure = &root;
                    continue;
                }

                next->failure = now->failure->next[c];
                for (auto s : next->failure->output) {
                    next->output.emplace_back(s);
                }
            }
        }
    }

    std::vector<std::pair<int, std::string>> find(const std::string &text) {
        std::vector<std::pair<int, std::string>> ans;
        auto now = &root;
        for (int i = 0; i < text.size(); ++i) {
            while (now != &root and now->next[text[i]] == nullptr) {
                now = now->failure;
            }

            if (now->next[text[i]] == nullptr) {
                continue;
            }

            now = now->next[text[i]];

            for (int j = 0; j < now->output.size(); ++j) {
                ans.emplace_back(make_pair(i - now->output[j].size() + 1, now->output[j]));
            }
        }
        return ans;
    }

};

using namespace std;

int main() {
    cin.tie(0);
    ios::sync_with_stdio(false);

    string S;
    cin >> S;
    int M;
    cin >> M;
    
    AhoCorasick ac;
    FOR(i, 0, M) {
        string C;
        cin >> C;
        ac.insert(C);
    }
    ac.build();

    print(ac.find(S).size());

    return 0;
}
0