結果

問題 No.430 文字列検索
コンテスト
ユーザー veqcc
提出日時 2019-08-10 17:29:42
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,761 bytes
コンパイル時間 512 ms
コンパイル使用メモリ 71,436 KB
実行使用メモリ 6,816 KB
最終ジャッジ日時 2024-11-10 00:27:23
合計ジャッジ時間 15,984 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 13 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <string>
#include <vector>
using namespace std;

// Z Algorithm
//  最長共通prefix長列を返す
//  計算量: O(|S|)

// アルゴリズム
//  k > r           -> それまで計算したzの結果は利用できず、ナイーブに計算する
//  z[k-l] < r-k+1  -> S[0:r-l] == S[l:r] であり、kがすっぽり入っているので一発
//  z[k-l] >= r-k+1 -> S[0:r-k+1] == S[k:r] であるので、追加分を計算する

vector <int> ZAlgorithm(const string& S) {
    int N = S.size();
    vector<int> z(N);
    z[0] = -1;

    int l = 0, r = 0;
    for (int k = 1; k < N; k++) {
        if (k > r) {
            l = k;
            r = k - 1;
            z[k] = 0;
            for (int i = 0; k + i < N; i++) {
                if (S[i] == S[k + i]) {
                    z[k]++;
                    r++;
                } else {
                    break;
                }
            }
        } else if (z[k - l] < r - k + 1) {
            z[k] = z[k - l];
        } else if (z[k - l] >= r - k + 1) {
            l = k;
            z[k] = r - k + 1;
            for (int i = r - k + 1; k + i < N; i++) {
                if (S[i] == S[k + i]) {
                    z[k]++;
                    r++;
                } else {
                    break;
                }
            }
        }
    }

    return z;
}


// verified
//  https://yukicoder.me/problems/no/430
void yuki430() {
    string S, C;
    int M;
    cin >> S >> M;
    int ans = 0;
    while (M--) {
        cin >> C;
        vector <int> Z = ZAlgorithm(C + S);
        for (int i = C.size(); i < Z.size(); i++) {
            if (Z[i] >= C.size()) ans++;
        }
    }
    cout << ans << "\n";
}


int main() {
    yuki430();
    return 0;
}
0