結果

問題 No.430 文字列検索
ユーザー sekiya9311sekiya9311
提出日時 2016-10-05 11:37:39
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 19 ms / 2,000 ms
コード長 2,808 bytes
コンパイル時間 1,050 ms
コンパイル使用メモリ 112,992 KB
実行使用メモリ 8,320 KB
最終ジャッジ日時 2024-05-03 04:38:47
合計ジャッジ時間 1,923 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 16 ms
8,320 KB
testcase_02 AC 11 ms
6,940 KB
testcase_03 AC 9 ms
6,944 KB
testcase_04 AC 2 ms
6,944 KB
testcase_05 AC 2 ms
6,944 KB
testcase_06 AC 2 ms
6,944 KB
testcase_07 AC 2 ms
6,940 KB
testcase_08 AC 10 ms
6,944 KB
testcase_09 AC 2 ms
6,944 KB
testcase_10 AC 3 ms
6,940 KB
testcase_11 AC 17 ms
6,940 KB
testcase_12 AC 19 ms
7,296 KB
testcase_13 AC 18 ms
7,168 KB
testcase_14 AC 16 ms
6,944 KB
testcase_15 AC 14 ms
6,940 KB
testcase_16 AC 12 ms
6,940 KB
testcase_17 AC 12 ms
6,944 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <string>
#include <queue>
#include <stack>
#include <algorithm>
#include <list>
#include <vector>
#include <complex>
#include <utility>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <climits>
#include <bitset>
#include <ctime>
#include <map>
#include <unordered_map>
#include <set>
#include <unordered_set>
#include <cassert>
#include <cstddef>
#include <iomanip>
#include <numeric>
#include <tuple>
#include <sstream>
#include <fstream>
#include <random>

#define REP(i, n) for (int (i) = 0; (i) < (n); (i)++)
#define FOR(i, a, b) for (int (i) = (a); (i) < (b); (i)++)
#define RREP(i, a) for(int (i) = (a)-1; (i) >= 0; (i)--)
#define FORR(i, a, b) for(int (i) = (a)-1; (i) >= (b); (i)--)
#define PI acos(-1.0)
#define DEBUG(C) cerr << C << endl;
#define VI vector <int>
#define VII vector <VI>
#define VL vector <LL>
#define VLL vector <VL>
#define VD vector <double>
#define VDD vector <VD>
#define PII pair <int, int>
#define PDD pair <double, double>
#define PLL pair <LL, LL>
#define VPII vector <PII>
#define ALL(a) (a).begin(), (a).end()
#define SORT(a) sort(ALL(a))
#define REVERSE(a) reverse(ALL(a))
#define MP make_pair
#define EB emplace_back
#define FORE(a, b) for (auto &&a:b)
#define FIND(s, n) (s.find(n) != s.end())

using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
const int INF = 1e9;
const int MOD = INF + 7;
const LL LLINF = 1e18;

template<typename T = int>
class Trie {
private:
    struct Node {
        Node *nxt[26];
        T w = 0;
    };
    Node root;
public:
    Trie() {
        root = Node();
        for (int i = 0; i < 26; i++) {
            root.nxt[i] = nullptr;
        }
    }
    void insert(const string& str, T w = 1) {
        Node *now;
        now = &this->root;
        for (const char c : str) {
            if (now->nxt[c - 'A'] == nullptr) {
                now->nxt[c - 'A'] = new Node();
            }
            now = now->nxt[c - 'A'];
        }
        now->w = w;
    }
    void make(const vector <pair<string, T>>& v) {
        for (pair<string, T> &&p : v) {
            inesrt(p.first, p.second);
        }
    }
    int search(const string& str) {
        Node *now;
        now = &this->root;
        for (const char c : str) {
            if (now->nxt[c - 'A'] == nullptr) return 0;
            now = now->nxt[c - 'A'];
        }
        return now->w;
    }
};

char buf[50010];
int main(void) {
    Trie <int> t;
    scanf("%s", buf);
    string S = buf;
    int M; scanf("%d", &M);
    REP(i, M) {
        scanf("%s", buf);
        string s = buf;
        t.insert(s);
    }
    int ans = 0;
    REP(i, 10) {
        REP(j, (int)S.length() - i) {
            ans += t.search(S.substr(j, i + 1));
        }
    }
    cout << ans << endl;
    return 0;
}
0