Cod sursa(job #788671)

Utilizator wefgefAndrei Grigorean wefgef Data 15 septembrie 2012 15:55:31
Problema PScPld Scor 40
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.96 kb
#include <cassert>
#include <cstdio>

#include <algorithm>
using namespace std;

typedef long long int64;

const int MAX_N = 1000005;

int n;
char v[2 * MAX_N];
int d[2 * MAX_N];

void read() {
    assert(freopen("pscpld.in", "r", stdin));
    assert(freopen("pscpld.out", "w", stdout));

    char ch;

    v[++n] = '#';
    while (scanf(" %c", &ch) == 1) {
        v[++n] = ch;
        v[++n] = '#';
    }
}

void solve() {
    int best = 0;
    for (int i = 1; i <= n; ++i) {
        if (best + d[best] >= i)
            d[i] = min(best + d[best] - i, d[2 * best - i]);
        while (i - d[i] - 1 > 0 && i + d[i] + 1 <= n &&
               v[i - d[i] - 1] == v[i + d[i] + 1])
            ++d[i];
        if (i + d[i] >= best)
            best = i;
    }

    int64 sol = 0;
    for (int i = 1; i <= n; ++i)
        if ((i % 2) == 1)
            sol += d[i] / 2;
        else
            sol += (d[i] + 1) / 2;
    printf("%lld\n", sol);
}

int main() {
    read();
    solve();
}