[CF1168B] Good Triple

给出 (01) 串 (s)，求数对 ([l,r]) 个数，使得能找到至少一对 (x,k)，使 (1le x,k le |s|) 且 (lle x<x+2k le r) 且(s[x]=s[x+k]=s[x+2k])

Solution

考虑一个暴力，对于所有的 (l)，暴力找到最小的能满足条件的 (r)

容易证明 (r) 是关于 (l) 单调的，于是倒序扫描 (l) 即可

#include <bits/stdc++.h>
using namespace std;

#define int long long
const int N = 1000005;

char s[N];
int n,ans,r;

signed main() {
    cin>>s+1;
    n=strlen(s+1);
    r=n+1;
    for(int i=n;i>=1;--i) {
        for(int j=1;i+2*j<=n;j++) {
            if(s[i]==s[i+j]&&s[i]==s[i+2*j]) {
                r=min(r,i+2*j);
                break;
            }
        }
        ans+=max(n-r+1,0ll);
    }
    cout<<ans;
}