結果

問題 No.599 回文かい
ユーザー xuzijian629
提出日時 2018-09-14 14:07:50
言語 C++11
(gcc 4.8.5)
結果
AC  
実行時間 1,373 ms
コード長 3,386 Byte
コンパイル時間 1,315 ms
使用メモリ 1,944 KB
最終ジャッジ日時 2019-10-23 00:03:09

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
0_0.txt AC 3 ms
1,552 KB
0_1.txt AC 4 ms
1,556 KB
0_2.txt AC 3 ms
1,556 KB
0_3.txt AC 3 ms
1,556 KB
1_0.txt AC 6 ms
1,576 KB
1_1.txt AC 5 ms
1,576 KB
1_2.txt AC 6 ms
1,584 KB
1_3.txt AC 6 ms
1,588 KB
1_4.txt AC 6 ms
1,584 KB
1_5.txt AC 3 ms
1,580 KB
2_0.txt AC 639 ms
1,816 KB
2_1.txt AC 379 ms
1,760 KB
2_2.txt AC 714 ms
1,832 KB
2_3.txt AC 415 ms
1,780 KB
2_4.txt AC 1,068 ms
1,912 KB
2_5.txt AC 1,234 ms
1,936 KB
2_6.txt AC 1,203 ms
1,920 KB
2_7.txt AC 1,373 ms
1,944 KB
3_0.txt AC 4 ms
1,576 KB
3_1.txt AC 4 ms
1,572 KB
3_2.txt AC 5 ms
1,576 KB
evil_0.txt AC 892 ms
1,880 KB
テストケース一括ダウンロード

ソースコード

diff #
#include <iostream>
#include <vector>
#include <algorithm>
#include <cmath>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <unordered_map>
#include <chrono>
#include <random>
#include <iterator>
#include <functional>
#include <utility>
#include <cassert>
#pragma GCC optimize("O3")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
#pragma comment(linker, "STACK:36777216")
using namespace std;
using i64 = int64_t;
constexpr i64 MOD = 1e9 + 7;
mt19937 rnd(chrono::steady_clock::now().time_since_epoch().count());
using vi = vector<i64>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using ii = pair<i64, i64>;
using i128 = __int128_t;

i64 modpow(i128 a, i128 n, i128 mod) {
    if (n == 0) return 1;
    if (n % 2 == 0) {
        i128 t = modpow(a, n / 2, mod);
        return t * t % mod;
    }
    return a * modpow(a, n - 1, mod) % mod;
}

i128 modinv(i128 n, i128 mod) {
    return modpow(n, mod - 2, mod);
}

bool is_prime(i128 n, int k) {
    if (n == 2) return true;
    if (n < 2 || n % 2 == 0) return false;
    i128 d = n - 1;
    while (d % 2 == 0) {
        d /= 2;
    }
    for (int i = 0; i < k; i++) {
        i128 a = rnd() % (n - 2) + 1;
        i128 t = d;
        i128 y = modpow(a, t, n);
        while (t != n - 1 && y != 1 && y != n - 1) {
            y = modpow(y, 2, n);
            t *= 2;
        }
        if (y != n - 1 && t % 2 == 0) {
            return false;
        }
    }
    return true;
}

i64 gen_prime(int bit) {
    assert(bit > 32);
    while (1) {
        i128 d = (i128(rnd()) << (bit - 32)) + rnd();
        d |= 1;
        if (is_prime(d, 50)) {
            return d;
        }
    }
}

class RollingHash {
    i128 hash[202020];
    i128 pows[202020];
    i128 p, m;
public:
    RollingHash(string s, i64 p = gen_prime(40), i64 m = gen_prime(50)) {
        assert(s.size() < 202020);
        hash[0] = 1;
        pows[0] = 1;
        for (int i = 0; i < s.size(); i++) {
            hash[i + 1] = (hash[i] * p + s[i]) % m;
            pows[i + 1] = pows[i] * p % m;
        }
        this->p = p;
        this->m = m;
    }
    
    i128 encode(int l, int r) {
        // inclusive
        return ((hash[r + 1] - hash[l] * pows[r - l + 1]) % m + m) % m;
    }
    
    bool eq(int l0, int r0, int l1, int r1) {
        return encode(l0, r0) == encode(l1, r1);
    }
};

int main() {
    string s;
    cin >> s;
    int n = s.size();
    RollingHash hash(s);
    if (n % 2) {
        vi dp(n / 2 + 1, 1);
        for (int i = 1; i <= n / 2; i++) {
            int l = n / 2 - i;
            int r = n / 2 + i;
            for (int d = 0; d < i; d++) {
                if (hash.eq(l, l + d, r - d, r)) {
                    assert(i - 1 - d >= 0);
                    dp[i] += dp[i - 1 - d];
                    dp[i] %= MOD;
                }
            }
        }
        cout << dp[n / 2] << endl;
    } else {
        vi dp(n / 2 + 1, 1);
        for (int i = 1; i <= n / 2; i++) {
            int l = n / 2 - i;
            int r = n / 2 - 1 + i;
            assert(r > l);
            for (int d = 0; d < i; d++) {
                if (hash.eq(l, l + d, r - d, r)) {
                    assert(i - 1 - d >= 0);
                    dp[i] += dp[i - 1 - d];
                    dp[i] %= MOD;
                }
            }
        }
        cout << dp[n / 2] << endl;
    }
}
0