【题目链接】
【思路要点】
- 在每两个字符间插入#(例如:“AAB”\(\Rightarrow\)“#A#A#B#”)。
- 用Manacher算法求出上述字符串每个字符为中心的最长回文子串的长度\(l_i\)。
- 记\(len_i=\frac{l_i+1}{2}\),不难发现\(len_i\)即为只算一边的字符时最长回文子串的长度,也即回文串的个数。
- 现在考虑每一个询问\([L,R]\),我们将其变为新字符串中的询问\([L*2-1,R*2+1]\),然后仅统计长度为奇数的回文子串的个数。
- 记原询问的答案为\(Ans\),新询问的答案为\(New\),有\(Ans=\frac{New-(R-L+2)}{2}\)(除去\(R-L+2\)个单独的“#”,每一个原询问的字符串S均对应了新询问中的两个字符串)。
- 考虑每个回文中心\(i\),若\(i-len_i+1≥L\)且\(i+len_i-1≤R\),那么它对询问答案的贡献显然是\(len_i\),否则,以\(i\)为中心的一部分回文串将超出这个区间的范围,我们称\(i\)被“限制了”。
- 记\(Mid=\frac{L+R}{2}\),显然\([L,Mid]\)中的位置只有可能被\(L\)“限制”,而\([Mid+1,R]\)中的位置只有可能被\(R\)“限制”。
- 将询问拆成\([L,Mid]\)、\([Mid+1,R]\)两部分,分别处理,由于处理方式对称,下面我们仅讨论\([L,Mid]\)的部分。
- 我们要求的实际上是\(\sum_{i=L}^{Mid}([i-len_i+1≥L]*len_i+[i-len_i+1<L]*(i-L+1))\)。
- 将询问离线,按照\(L\)排序,不难用树状数组求得上述式子的值。
- 时间复杂度\(O((N+Q)LogN)\)。
【代码】
#include<bits/stdc++.h> using namespace std; const int MAXN = 300005; template <typename T> void chkmax(T &x, T y) {x = max(x, y); } template <typename T> void chkmin(T &x, T y) {x = min(x, y); } template <typename T> void read(T &x) { x = 0; int f = 1; char c = getchar(); for (; !isdigit(c); c = getchar()) if (c == '-') f = -f; for (; isdigit(c); c = getchar()) x = x * 10 + c - '0'; x *= f; } template <typename T> void write(T x) { if (x < 0) x = -x, putchar('-'); if (x > 9) write(x / 10); putchar(x % 10 + '0'); } template <typename T> void writeln(T x) { write(x); puts(""); } struct querys {int pos, home; }; vector <querys> l[MAXN], r[MAXN]; char s[MAXN]; int n, q, len[MAXN]; long long ans[MAXN]; struct BinaryIndexTree { int n; long long a[MAXN]; void init(int x) { n = x; memset(a, 0, sizeof(a)); } void modify(int pos, int x) { for (int i = pos; i <= n; i += i & -i) a[i] += x; } long long query(int pos) { long long ans = 0; for (int i = pos; i >= 1; i -= i & -i) ans += a[i]; return ans; } } Inside, Sum, Cnt; void solvel() { Inside.init(n); Sum.init(n), Cnt.init(n); static vector <int> dis[MAXN]; for (int i = 1; i <= n; i++) { Inside.modify(i, len[i]); dis[i - len[i] + 1].push_back(i); } for (int i = 1; i <= n; i++) { for (unsigned j = 0; j < l[i].size(); j++) { long long tans = 0; tans += Inside.query(l[i][j].pos); tans += Sum.query(l[i][j].pos) - Cnt.query(l[i][j].pos) * (i - 1); ans[l[i][j].home] += tans; } for (unsigned j = 0; j < dis[i].size(); j++) { int tmp = dis[i][j]; Inside.modify(tmp, -len[tmp]); Sum.modify(tmp, tmp); Cnt.modify(tmp, 1); } Sum.modify(i, -i); Cnt.modify(i, -1); } } void solver() { Inside.init(n); int m = n + 1; Sum.init(n), Cnt.init(n); static vector <int> dis[MAXN]; for (int i = 1; i <= n; i++) { Inside.modify(m - i, len[i]); dis[i + len[i] - 1].push_back(i); } for (int i = n; i >= 1; i--) { for (unsigned j = 0; j < r[i].size(); j++) { long long tans = 0; tans += Inside.query(m - r[i][j].pos); tans += Cnt.query(m - r[i][j].pos) * (i + 1) - Sum.query(m - r[i][j].pos); ans[r[i][j].home] += tans; } for (unsigned j = 0; j < dis[i].size(); j++) { int tmp = dis[i][j]; Inside.modify(m - tmp, -len[tmp]); Sum.modify(m - tmp, tmp); Cnt.modify(m - tmp, 1); } Sum.modify(m - i, -i); Cnt.modify(m - i, -1); } } int main() { scanf("%s", s + 1); n = strlen(s + 1); for (int i = n; i >= 1; i--) s[i * 2] = s[i]; for (int i = 0; i <= n; i++) s[i * 2 + 1] = '#'; s[0] = '$', s[n * 2 + 2] = '%'; n = n * 2 + 1; int pos = 0, home = 0; for (int i = 1; i <= n; i++) { if (i <= pos) { int j = 2 * home - i; len[i] = min(len[j], pos - i); } while (s[i + len[i]] == s[i - len[i]]) len[i]++; if (i + len[i] - 1 >= pos) { pos = i + len[i] - 1; home = i; } } read(q); for (int i = 1; i <= q; i++) { int x, y, z; read(x), read(y); ans[i] = -(y - x + 2); x = 2 * x - 1; y = 2 * y + 1; z = (x + y) / 2; l[x].push_back((querys) {z, i}); r[y].push_back((querys) {z + 1, i}); } solvel(); solver(); for (int i = 1; i <= q; i++) writeln(ans[i] / 2); return 0; }