#include <bits/stdc++.h>
using namespace std;
// Z-function: 计算字符串 s 的 Z-array
vector<int> z_function(const string& s) {
int n = s.size();
vector<int> z(n); // z[i] 表示 s 的前缀和从 i 开始的子串的最长公共前缀长度
int l = 0, r = 0; // l, r 表示当前 Z-box 的左右边界
for (int i = 1; i < n; i++) {
if (i <= r)
z[i] = min(r - i + 1, z[i - l]); // 在当前 Z-box 内,使用之前计算的值
while (i + z[i] < n && s[z[i]] == s[i + z[i]])
z[i]++; // 尝试扩展当前的 Z-box
if (i + z[i] - 1 > r) {
l = i; // 更新 Z-box 的左右边界
r = i + z[i] - 1;
}
}
return z; // 返回 Z-array
}
int main() {
ios::sync_with_stdio(0);
cin.tie(0),cout.tie(0);
int t;
cin >> t; // 读取测试用例数
while (t--) {
string s;
cin >> s; // 读取当前测试用例的字符串
int n = s.size(); // 获取字符串长度
// nxt_nona[i] 表示从位置 i 开始最近的一个非 'a' 字符的位置
vector<int> nxt_nona(n + 1, n); // 初始为 n 表示没有非 'a' 的字符
for (int i = n - 1; i >= 0; i--) {
nxt_nona[i] = (s[i] == 'a') ? nxt_nona[i + 1] : i;
// 如果 s[i] 是 'a',则继承下一个位置的值;否则更新为 i
}
if (nxt_nona[0] == n) { // 如果字符串全为 'a'
cout << n - 1 << '\n'; // 输出 n-1,表示可以切分的方式数
continue; // 处理下一个测试用例
}
int p0 = nxt_nona[0]; // 找到第一个非 'a' 字符的位置
string T = s.substr(p0); // 提取从 p0 开始的子串 T
auto z = z_function(T); // 计算子串 T 的 Z-array
long long ans = 0; // 最终答案
for (int len = 1; len <= n - p0; len++) { // 枚举可能的子串长度 len
int min_gap = p0; // 初始化最小间隙
bool valid = true; // 标记当前子串长度是否有效
int last_pos = p0 + len - 1; // 当前子串的结束位置
for (int j = nxt_nona[p0 + len]; j < n; j = nxt_nona[j + len]) {
// 遍历从下一个非 'a' 字符开始的可能分割点
if (z[j - p0] < len) { // 如果 Z-array 的值小于 len,则不能匹配
valid = false; // 当前长度的子串不合法
break; // 退出循环
}
min_gap = min(min_gap, j - last_pos - 1); // 更新最小间隙
last_pos = j + len - 1; // 更新最后一个位置
}
if (valid) ans += min_gap + 1; // 如果当前长度的子串合法,累加答案
}
cout << ans << '\n'; // 输出当前测试用例的结果
}
return 0; // 程序结束
}
D. “a“ String Problem
于 2024-08-03 08:55:03 首次发布