結果

問題 No.465 PPPPPPPPPPPPPPPPAPPPPPPPP
ユーザー pekempeypekempey
提出日時 2016-12-16 08:26:43
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 2,431 bytes
コンパイル時間 1,692 ms
コンパイル使用メモリ 176,128 KB
実行使用メモリ 39,352 KB
最終ジャッジ日時 2024-11-30 10:17:06
合計ジャッジ時間 27,324 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
13,640 KB
testcase_01 AC 2 ms
26,224 KB
testcase_02 AC 2 ms
13,640 KB
testcase_03 AC 2 ms
26,256 KB
testcase_04 AC 2 ms
13,632 KB
testcase_05 AC 18 ms
26,688 KB
testcase_06 AC 81 ms
26,504 KB
testcase_07 AC 23 ms
26,208 KB
testcase_08 AC 94 ms
25,900 KB
testcase_09 TLE -
testcase_10 TLE -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 TLE -
testcase_16 TLE -
testcase_17 TLE -
testcase_18 TLE -
testcase_19 AC 38 ms
19,876 KB
testcase_20 TLE -
testcase_21 TLE -
32_ratsliveonnoevilstar.txt AC 71 ms
39,848 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

bool isPalindrome(const string &s, int l, int r) {
    for (int i = 0; i < r - l; i++) {
        if (s[l + i] != s[r - 1 - i]) {
            return false;
        }
    }
    return true;
}

bool isPalTwo(const string &s, int k, const vector<int> &zF) {
    assert(k >= 2);
    // evil solution
    for (int i = 1; i < min(100, k); i++) {
        if (zF[s.size() * 2 + 1 - i] >= i && isPalindrome(s, i, k)) {
            return true;
        }
    }
    /*
    for (int i = 1; i < r - l; i++) {
        if (isPalindrome(s, l, l + i) && isPalindrome(s, l + i, r)) {
            return true;
        }
    }
    */
    return false;
}

vector<int> zAlgorithm(string s) {
    vector<int> z(s.size());
    z[0] = s.size();
    int i = 1;
    int j = 0;
    while (i < s.size()) {
        while (i + j < s.size() && s[i + j] == s[j]) {
            j++;
        }
        z[i++] = j;
        if (j == 0) {
            continue;
        }
        j--;
        int k = 1;
        while (i < s.size() && z[k] < j) {
            z[i++] = z[k++];
            j--;
        }
    }
    return z;
}

int main() {
    string s;
    cin >> s;
    string t = s;
    reverse(t.begin(), t.end());
    int n = s.size();

    vector<bool> bad(n);
    vector<int64_t> cnt(n + 2);
    vector<int> z = zAlgorithm(s);
    vector<int> zF = zAlgorithm(s + "$" + t);
    vector<int> zB = zAlgorithm(t + "$" + s);

    for (int i = 1; i < n; i++) {
        if (bad[i]) {
            continue;
        }
        if (zF[n * 2 + 1 - i] >= i) {
            for (int j = 2; i * j < n; j++) {
                if (z[i * (j - 1)] < i) {
                    break;
                }
                cnt[i * j + 1] += j - 1;
                bad[i * j] = true;
            }
        }
    }
    for (int i = 2; i <= n - 2; i++) {
        if (bad[i]) {
            continue;
        }
        // TODO: more faster
        if (isPalTwo(s, i, zF)) {
            for (int j = 1; i * j < n; j++) {
                if (z[i * (j - 1)] < i) {
                    break;
                }
                cnt[i * j + 1] += j;
                bad[i * j] = true;
            }
        }
    }

    for (int i = 0; i < n - 1; i++) {
        cnt[i + 1] += cnt[i];
    }

    int64_t ans = 0;
    for (int i = 1; i < n; i++) {
        if (zB[n * 2 + 1 - i] >= i) {
            ans += cnt[n - i];
        }
    }
    cout << ans << endl;
}
0