Cod sursa(job #2232024)

Utilizator DenisacheDenis Ehorovici Denisache Data 17 august 2018 00:00:38
Problema PScPld Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.65 kb
#include <cstdio>
#include <cstring>

using namespace std;

int d[2000005];
char s[2000005];

inline int min(const int& a, const int& b) {
	return (a < b) ? a : b;
}

int main() {
	freopen("pscpld.in", "r", stdin);
	freopen("pscpld.out", "w", stdout);
	
	int n = 0;
	s[n++] = '+';
	while (scanf("%c", &s[n++]) > 0)
		s[n++] = '+';

	long long res = 0;
	int r = 1;

	for (int i = 1; i < n - 1; ++i) {
		if (r + d[r] > i)
			d[i] = min(d[2 * r - i], r + d[r] - i);

		while (0 <= i - d[i] && i + d[i] < n && s[i - d[i]] == s[i + d[i]]) {
			++d[i];
			r = i;
		}

		res += 1LL * d[i] / 2;
	}

	printf("%lld", res);

	return 0;
}