洛谷P1872 回文串计数

tags

字符串哈希 枚举

中文题面

题目描述

小 aa 虽然是一名理科生，但他常常称自己是一名真正的文科生。不知为何，他对于背诵总有一种莫名其妙的热爱，这也促使他走向了以记忆量大而闻名的生物竞赛。然而，他很快发现这并不能满足他热爱背诵的心，但是作为一名强大的 OIER，他找到了这么一个方法——背诵基因序列。然而这实在是太困难了，小啊感觉有些招架不住。

不过他发现，如果他能事先知道这个序列里有多少对互不相交的回文串，他或许可以找到记忆的妙法。为了进一步验证这个想法，小 aa 决定选取一个由小写字母构成的字符串 SSSS 来实验。由于互不相关的回文串实在过多，他很快就数晕了。不过他相信，在你的面前这个问题不过是小菜一碟。

对于字符串 SSSS，设其长度为 Len，那么下文用 Si​ 表示 SSSS 中第 i 个字符（1≤i≤Len​）。

S[i,j] 表示 SSSS 的一个子串，S[i,j]=SiSi+1Si+2⋯Sj−2Sj−1Sj，比如当 SSSS 为 abcgfd 时，S[2,5] 就是 bcgf，S[1,5] 是 abcgf。

当一个串被称为一个回文串当且仅当将这个串反写后与原串相同，如abcba。

考虑一个四元组 (l,r,L,R)，当 S[l,r] 和 S[L,R]均为回文串时，且满足 1≤l≤r<L≤R≤Len 时，我们称 S[l,r] 和 S[L,R] 为一对互不相交的回文串。即本题所求，也即为这种四元组的个数。两个四元组相同当且仅当对应的 l,r,L,R 都相同。

输入格式

输入仅一行，为字符串 SSSS，保证全部由小写字母构成，由换行符标志结束。

50 %的数据满足 SSSS 的长度不超过 200；

100 ％的数据满足 SSSS 的长度不超过 2000。
输出格式

仅一行，为一个整数，表示互不相关的回文串的对数。
输入输出样例
输入 #1

aaa

输出 #1

5

说明/提示
【样例数据说明】

SS=“aaa”，SS 的任意一个字符串均为回文串，其中总计有 5 对互不相关的回文串：

(1,1,2,2)，(1,1,2,3)，(1,1,3,3)，(1,2,3,3)，(2,2,3,3)。

思路

为了快速找出字符串中的子字符串，进行字符串哈希预处理（前缀哈希与后缀哈希），由于数据量不是很大，枚举可能的开头i，结尾j，如果为回文串则记录。
对于一个开头i结尾j的回文串，它能贡献的四元组数量为开头k(k > j)的回文串数量，则我们在记录回文串的过程中顺便记录某点起始的回文串数量，再逆向dp得到某点之后的回文串数量，本题得解。

代码

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define ull unsigned long long

const int n = 2000, base = 131;
ull head[n + 5];
ull back[n + 5];
ull p[n + 5];
int oj[2005][2005];
int cnt[2005];

ull idx(char& x) {
	return x - 'a' + 1;
}
ull geth(int l, int r) {
	return head[r] - head[l - 1] * p[r - l + 1];
}
ull getb(int l, int r) {
	return back[l] - back[r + 1] * p[r - l + 1];
}

signed main() {
	ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
	head[0] = 0;;
	p[0] = 1;
	int res = 0;
	string s;
	cin >> s;
	int n = s.length();
	back[n + 1] = 0;
	for (int i = 1; i <= n; i++) {
		head[i] = head[i - 1] * base + idx(s[i - 1]);
		back[n + 1 - i] = back[n + 2 - i] * base + idx(s[n - i]);
		p[i] = p[i - 1] * base;
	}
	for (int i = 1; i <= n; i++) {
		for (int j = i; j <= n; j++) {
			if (geth(i, j) == getb(i, j)) oj[i][j] = 1, cnt[i]++;
		}
	}
	for (int i = n - 1; i >= 1; i--) {
		cnt[i] += cnt[i + 1];
	}
	for (int i = 1; i <= n; i++) {
		for (int j = i; j <= n; j++) {
			if (oj[i][j]) {
				res += cnt[j + 1];
			}
		}
	}
	cout << res;
	return 0;
}