Submission #617279

#TimeUsernameProblemLanguageResultExecution timeMemory
617279maximath_1Palinilap (COI16_palinilap)C++11
100 / 100
126 ms31916 KiB
#include <bits/stdc++.h>
using namespace std;

#define ll long long
const int MX = 1e5 + 5;
const ll mod = 420691273;
const ll mod2 = 1e9 + 9;
const ll pp = 37;
const ll pp2 = 31;

string s, sr;
int n;
ll pwp[MX][2], ipwp[MX][2], hsh[MX][2], hshr[MX][2];

ll pw(ll b, ll e, ll m){
	ll r = 1ll; b %= m;
	for(; e > 0; e /= 2, (b *= b) %= m)
		if(e % 2) (r *= b) %= m;
	return r;
}

bool check_palindrome(int lf, int cl, int cr, int rg){
	rg = n - 1 - rg; cr = n - 1 - cr;
	if(lf > cl || rg > cr || lf < 0 || rg < 0) return 0;

	bool ok = 1;
	for(int i = 0; i < 2; i ++){
		ll hashL = hsh[cl][i] - (lf == 0 ? 0 : hsh[lf - 1][i]); if(hashL < 0) hashL += (i ? mod2 : mod);
		(hashL *= ipwp[lf][i]) %= (i ? mod2 : mod);
		ll hashR = hshr[cr][i] - (rg == 0 ? 0 : hshr[rg - 1][i]); if(hashR < 0) hashR += (i ? mod2 : mod);
		(hashR *= ipwp[rg][i]) %= (i ? mod2 : mod);
		ok &= (hashL == hashR);
	}

	return ok;
}

ll add[MX][30]; // change s[i] to j + 'a', how many palindromes are added
ll delRunning[MX]; ll delCap[MX];
/*
 a  b  c  d  d  c  b  a   |  a  b  c  b  a
 1  2  3  4  4  3  2  1   |  1  2  0  2  1
changing s[i] will delete palindromes according to above
for left side, represent as i - x, x constant (delRuuning = 1, delCap = -x)
for right, represent as x - i, x constant (delRunning = -1, delCap = x)
when added, will become in the form of i * delRunning + delCap
is range update point query, with all queries processed at the end -> offline prefix difference
*/

int main(){
	cin >> s;
	sr = s; reverse(sr.begin(),sr.end());
	n = s.size();

	pwp[0][0] = 1;
	for(int i = 1; i < MX; i ++)
		pwp[i][0] = (pwp[i - 1][0] * 1ll * pp) % mod;
	ipwp[MX - 1][0] = pw(pwp[MX - 1][0], mod - 2, mod);
	for(int i = MX - 2; i >= 0; i --)
		ipwp[i][0] = (ipwp[i + 1][0] * 1ll * pp) % mod;
	pwp[0][1] = 1;
	for(int i = 1; i < MX; i ++)
		pwp[i][1] = (pwp[i - 1][1] * 1ll * pp2) % mod2;
	ipwp[MX - 1][1] = pw(pwp[MX - 1][1], mod2 - 2, mod2);
	for(int i = MX - 2; i >= 0; i --)
		ipwp[i][1] = (ipwp[i + 1][1] * 1ll * pp2) % mod2;

	for(int i = 0; i < n; i ++){
		for(int j = 0; j < 2; j ++){
			hsh[i][j] = (pwp[i][j] * 1ll * (s[i] - 'a')) % (j ? mod2 : mod);
			hshr[i][j] = (pwp[i][j] * 1ll * (sr[i] - 'a')) % (j ? mod2 : mod);

			if(i > 0){
				hsh[i][j] += hsh[i - 1][j]; if(hsh[i][j] >= (j ? mod2 : mod)) hsh[i][j] -= (j ? mod2 : mod);
				hshr[i][j] += hshr[i - 1][j]; if(hshr[i][j] >= (j ? mod2 : mod)) hshr[i][j] -= (j ? mod2 : mod);
			}
		}
	}

	ll base_answer = 0ll;
	for(int cl = 0; cl < n; cl ++){
		for(int cr = cl; cr <= cl + 1; cr ++){
			int lf = 0, rg = min(cl + 1, n - cr), rs = lf;
			for(int md; lf <= rg;){
				md = (lf + rg + 1) / 2;
				if(check_palindrome(cl - md + 1, cl, cr, cr + md - 1)){
					rs = md;
					lf = md + 1;
				}else{
					rg = md - 1;
				}
			}
			base_answer += rs;

			// process deleting
			if(cl == cr){
				// odd case
				delRunning[cl - rs + 1] ++; delRunning[cl] --; // left side
				delCap[cl - rs + 1] -= cl - rs; delCap[cl] += cl - rs;

				delRunning[cr + 1] --; delRunning[cr + rs] ++; // right side
				delCap[cr + 1] += cr + rs; delCap[cr + rs] -= cr + rs;
			}else{
				// even case
				delRunning[cl - rs + 1] ++; delRunning[cl + 1] --; // left side
				delCap[cl - rs + 1] -= cl - rs; delCap[cl + 1] += cl - rs;

				delRunning[cr] --; delRunning[cr + rs] ++; // right side
				delCap[cr] += cr + rs; delCap[cr + rs] -= cr + rs;
			}

			// process adding
			int xl = cl - rs, xr = cr + rs; rs ++;
			if(xl < 0 || xr >= n) continue;
			lf = 0, rg = min(cl + 1, n - cr) - rs; rs = lf;
			for(int md; lf <= rg;){
				md = (lf + rg + 1) / 2;
				if(check_palindrome(xl - md, xl - 1, xr + 1, xr + md)){
					rs = md;
					lf = md + 1;
				}else{
					rg = md - 1;
				}
			}

			add[xl][s[xr] - 'a'] += rs + 1;
			add[xr][s[xl] - 'a'] += rs + 1;
		}
	}

	ll best_change = 0ll;
	ll run = 0ll, cap = 0ll;
	for(int i = 0; i < n; i ++){
		run += delRunning[i];
		cap += delCap[i];
		for(int j = 0; j < 26; j ++) if(s[i] - 'a' != j){
			best_change = max(best_change, add[i][j] - (run * 1ll * i + cap));
		}
	}

	cout << base_answer + best_change << endl;
	return 0;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...