oj.uz

Submission #838755

# Submission time Handle Problem Language Result Execution time Memory
838755 2023-08-27T17:15:58 Z popovicirobert Palindromes (APIO14_palindrome) C++14
23 / 100
 164 ms 44148 KB 
#include <bits/stdc++.h>

using namespace std;

using ull = unsigned long long;

constexpr int BASE = (int) 1e9 + 7;
constexpr int MAXN = (int) 3e5;

char s[MAXN + 5];
ull pref_hash[MAXN + 5];
ull power[MAXN + 5];

inline ull Get(int l, int r) {
    return pref_hash[r] - pref_hash[l - 1] * power[r - l + 1];
}

char str[2 * MAXN + 5];
int len[2 * MAXN + 5];

int Build(char* s, int n) {
    power[0] = 1;
    for (int i = 1; i <= n; i++) {
        pref_hash[i] = pref_hash[i - 1] * BASE + s[i] - 'a' + 1;
        power[i] = power[i - 1] * BASE;
    }

    int size = 0;
    str[++size] = '*';
    for (int i = 1; i <= n; i++) {
        str[++size] = s[i];
        str[++size] = '*';
    }

    return size;
}

void Manacher(char* str, int size) {
    int pos = 1;
    len[1] = 1;

    for (int i = 2; i <= size; i++) {
        len[i] = 1;
        if (pos + len[pos] >= i) {
            len[i] = min(pos + len[pos] - i, len[2 * pos - i]);
        }

        while (i - len[i] >= 1 && i + len[i] <= size && str[i - len[i]] == str[i + len[i]]) {
            len[i]++;
        }

        if (pos + len[pos] < i + len[i]) {
            pos = i;
        }
    }
}

unordered_map<ull, int> nodeId;
vector<vector<int>> g;

vector<int> nodeFreq(1);
vector<int> nodeLen(1);

void BuildGraph(int size) {

    int ID = 0;

    for (int i = 1; i <= size; i++) {
        int l = len[i];
        if (i % 2 == l % 2) {
            l--;
        }

        int prevId = 0;

        while (l > 0) {
            ull currHash;
            int mid = i / 2;

            if (i % 2 == 1) {
                currHash = Get(mid - l / 2 + 1, mid + l / 2);
            } else {
                currHash = Get(mid - l / 2, mid + l / 2);
            }

            int& currId = nodeId[currHash];
            if (currId == 0) {
                currId = ++ID;
                nodeLen.push_back(l);
                nodeFreq.push_back(0);
            }

            if (currId >= (int)g.size()) {
                g.resize(currId + 1);
            }

            if (prevId == 0) {
                nodeFreq[currId]++;
            } else {
                g[prevId].push_back(currId);
            }

            if (currId != ID) {
                break;
            }

            prevId = currId;
            l -= 2;
        }
    }
}


int main() {
#ifdef HOME
    ifstream cin("input.in");
    ofstream cout("output.out");
#endif
    ios::sync_with_stdio(false);
    cin.tie(0), cout.tie(0);

    cin >> (s + 1);
    int n = strlen(s + 1);

    int size = Build(s, n);

    Manacher(str, size);

    // for (int i = 1; i <= size; i++) {
    //     cerr << (char)str[i] << " ";
    // }
    // cerr << "\n";
    // for (int i = 1; i <= size; i++) {
    //     cerr << len[i] << " ";
    // }
    // cerr << "\n";

    BuildGraph(size);

    int nodes = (int)g.size() - 1;
    vector<int> order(nodes);
    iota(order.begin(), order.end(), 1);
    sort(order.begin(), order.end(), [](const int& a, const int& b) {
        return nodeLen[a] > nodeLen[b];
    });

    long long answer = 0;
    for (auto node : order) {
        // cerr << nodeFreq[node] << " " << nodeLen[node] << "\n";
        answer = max(answer, 1LL * nodeFreq[node] * nodeLen[node]);
        for (auto son : g[node]) {
            nodeFreq[son] += nodeFreq[node];
        }
    }

    cout << answer;

    return 0;
}

Subtask #1
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 340 KB Output is correct
2 Correct 1 ms 340 KB Output is correct
3 Correct 1 ms 340 KB Output is correct
4 Correct 1 ms 328 KB Output is correct
5 Correct 1 ms 340 KB Output is correct
6 Correct 1 ms 340 KB Output is correct
7 Correct 1 ms 340 KB Output is correct
8 Correct 1 ms 340 KB Output is correct
9 Correct 1 ms 340 KB Output is correct
10 Correct 1 ms 340 KB Output is correct
11 Correct 0 ms 336 KB Output is correct
12 Correct 1 ms 340 KB Output is correct
13 Correct 1 ms 340 KB Output is correct
14 Correct 1 ms 340 KB Output is correct
15 Correct 1 ms 340 KB Output is correct
16 Correct 0 ms 328 KB Output is correct
17 Correct 1 ms 340 KB Output is correct
18 Correct 0 ms 340 KB Output is correct
19 Correct 0 ms 340 KB Output is correct
20 Correct 1 ms 340 KB Output is correct
21 Correct 1 ms 328 KB Output is correct
22 Correct 1 ms 340 KB Output is correct
23 Correct 1 ms 340 KB Output is correct
24 Correct 1 ms 340 KB Output is correct
25 Correct 1 ms 340 KB Output is correct
26 Correct 1 ms 340 KB Output is correct
27 Correct 1 ms 340 KB Output is correct
28 Correct 1 ms 340 KB Output is correct
29 Correct 1 ms 332 KB Output is correct
30 Correct 1 ms 340 KB Output is correct
31 Correct 1 ms 340 KB Output is correct
32 Correct 1 ms 340 KB Output is correct

Subtask #2
15.0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 512 KB Output is correct
2 Correct 1 ms 468 KB Output is correct
3 Correct 1 ms 468 KB Output is correct
4 Correct 1 ms 468 KB Output is correct
5 Correct 1 ms 468 KB Output is correct
6 Correct 1 ms 468 KB Output is correct
7 Correct 1 ms 468 KB Output is correct
8 Correct 1 ms 468 KB Output is correct
9 Correct 1 ms 468 KB Output is correct
10 Correct 1 ms 340 KB Output is correct
11 Correct 1 ms 340 KB Output is correct
12 Correct 1 ms 468 KB Output is correct

Subtask #3
0 / 24.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 1748 KB Output is correct
2 Correct 3 ms 1748 KB Output is correct
3 Correct 3 ms 1744 KB Output is correct
4 Correct 3 ms 1748 KB Output is correct
5 Correct 3 ms 1748 KB Output is correct
6 Correct 3 ms 1748 KB Output is correct
7 Correct 3 ms 1748 KB Output is correct
8 Correct 1 ms 596 KB Output is correct
9 Correct 1 ms 596 KB Output is correct
10 Incorrect 2 ms 852 KB Output isn't correct
11 Halted 0 ms 0 KB -

Subtask #4
0 / 26.0

# Verdict Execution time Memory Grader output
1 Correct 33 ms 14144 KB Output is correct
2 Correct 36 ms 14204 KB Output is correct
3 Correct 32 ms 14196 KB Output is correct
4 Correct 36 ms 14208 KB Output is correct
5 Correct 34 ms 14204 KB Output is correct
6 Correct 24 ms 11916 KB Output is correct
7 Correct 27 ms 11984 KB Output is correct
8 Correct 5 ms 3160 KB Output is correct
9 Correct 11 ms 5656 KB Output is correct
10 Incorrect 15 ms 7544 KB Output isn't correct
11 Halted 0 ms 0 KB -

Subtask #5
0 / 27.0

# Verdict Execution time Memory Grader output
1 Correct 150 ms 44148 KB Output is correct
2 Correct 161 ms 44140 KB Output is correct
3 Correct 164 ms 44136 KB Output is correct
4 Correct 164 ms 44116 KB Output is correct
5 Correct 153 ms 44132 KB Output is correct
6 Correct 134 ms 44136 KB Output is correct
7 Correct 132 ms 35672 KB Output is correct
8 Incorrect 14 ms 8696 KB Output isn't correct
9 Halted 0 ms 0 KB -