[CF1168B] Good Triple

给出 \(01\) 串 \(s\)，求数对 \([l,r]\) 个数，使得能找到至少一对 \(x,k\)，使 \(1\le x,k \le |s|\) 且 \(l\le 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;
}