D. “a“ String Problem

#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; // 程序结束
}