Cod sursa(job #2684328)

Utilizator TheGodFather2131Alexandru Miclea TheGodFather2131 Data 13 decembrie 2020 15:39:08
Problema PScPld Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.24 kb
//ALEXANDRU MICLEA

#include <vector>
#include <algorithm>
#include <string>
#include <string.h>
#include <cstring>
#include <queue>
#include <map>
#include <set>
#include <unordered_map>
#include <time.h>
#include <iomanip>
#include <deque>
#include <math.h>
#include <cmath>
#include <assert.h>
#include <stack>
#include <bitset>
#include <random>
#include <chrono>
#include <assert.h>

using namespace std;
using ll = long long;

#include <fstream>
//ifstream cin("input.in"); ofstream cout("output.out");
ifstream cin("pscpld.in"); ofstream cout("pscpld.out");

//VARIABLES

string s, aux;
int n;

//FUNCTIONS

ll manacher() {
    ll ans = 0;
    vector <int> d1(n);

    for (int i = 0, l = 0, r = -1; i < n; i++) {
        int k = (i > r) ? 1 : min(d1[l + r - i], r - i + 1);
        while (i - k >= 0 && i + k < n && s[i - k] == s[i + k]) k++;

        d1[i] = k--;

        if (i + k > r) {
            l = i - k;
            r = i + k;
        }

        ans += d1[i] / 2;
    }

    return ans;
}

//MAIN

int main() {

    cin >> aux;

    s = "#";

    for (auto& x : aux) {
        s += x;
        s += "#";
    }

    n = s.size();
    cout << manacher() << '\n';
    
    return 0;
}