Cod sursa(job #2684318)

Utilizator TheGodFather2131Alexandru Miclea TheGodFather2131 Data 13 decembrie 2020 15:26:38
Problema PScPld Scor 30
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.21 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 aux, s;
int d1[(int)2e6 + 5];

//FUNCTIONS

ll manacher(string s) {
    int n = s.size();
    int ans = 0;

    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 (0 <= i - k && 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 += "#";
    }
    
    cout << manacher(s) << '\n';
    
    return 0;
}