結果

問題 No.599 回文かい
ユーザー veqccveqcc
提出日時 2019-08-21 15:44:25
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 272 ms / 4,000 ms
コード長 2,109 bytes
コンパイル時間 1,094 ms
コンパイル使用メモリ 76,152 KB
実行使用メモリ 126,720 KB
最終ジャッジ日時 2024-10-09 04:29:31
合計ジャッジ時間 2,829 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,820 KB
testcase_02 AC 2 ms
6,820 KB
testcase_03 AC 2 ms
6,820 KB
testcase_04 AC 2 ms
6,820 KB
testcase_05 AC 2 ms
6,816 KB
testcase_06 AC 2 ms
6,816 KB
testcase_07 AC 2 ms
6,820 KB
testcase_08 AC 2 ms
6,820 KB
testcase_09 AC 2 ms
6,816 KB
testcase_10 AC 2 ms
6,820 KB
testcase_11 AC 2 ms
6,820 KB
testcase_12 AC 3 ms
6,820 KB
testcase_13 AC 3 ms
6,820 KB
testcase_14 AC 138 ms
102,528 KB
testcase_15 AC 9 ms
8,832 KB
testcase_16 AC 239 ms
111,232 KB
testcase_17 AC 272 ms
126,720 KB
testcase_18 AC 2 ms
6,816 KB
testcase_19 AC 2 ms
6,820 KB
testcase_20 AC 2 ms
6,816 KB
evil_0.txt AC 60 ms
44,928 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <string>
#include <vector>
typedef long long ll;
using namespace std;
const ll MOD = 1000000007LL;

// 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) {
            int i = 0;
            while (k + i < N && S[i] == S[k + i]) i++;
            l = k;
            r = k + i - 1;
            z[k] = i;
        } else if (z[k - l] < r - k + 1) {
            z[k] = z[k - l];
        } else if (z[k - l] >= r - k + 1) {
            int i = r - k + 1;
            while (k + i < N && S[i] == S[k + i]) i++;
            l = k;
            r = k + i - 1;
            z[k] = i;
        }
    }

    return z;
}


// verified (TLE)
//  https://yukicoder.me/problems/no/430
void yuki430() {
    string S, C;
    int M, ans = 0;
    cin >> S >> M;
    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";
}


// verified
//  https://yukicoder.me/problems/no/599
vector <ll> dp(10005);
vector <bool> used(10005, false);
ll dfs(const string &S, int L, int R) {
    if (R <= L) return 1;
    if (used[L]) return dp[L];
    string s = S.substr(L, R - L);
    vector <int> z = ZAlgorithm(s);
    ll ret = 1;
    int l = 1, r = s.size() - 1;
    while (l <= r) {
        if (z[r] == l) ret = (ret + dfs(S, L + l, L + r)) % MOD;
        l++; r--;
    }
    used[L] = true;
    return dp[L] = ret;
}
void yuki599() {
    string s;
    cin >> s;
    cout << dfs(s, 0, s.size()) << '\n';
}

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