【题解】
题意:输出所有符合条件的区间对(l,r)的个数。给定一个01串a,要求:在[l,r]内至少存在一对ai=ai+k=ai+2k。
思路:字符串从前跑到后,记录并不断更新最长可加前缀(不加算在内),每次加上最长可加前缀的长度即为答案。
【代码】
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=3e5+10;
char s[maxn];
int main()
{
scanf("%s",s+1);
int n=strlen(s+1);
ll ans=0;
for(int i=1,l=0;i<=n;i++){
for(int j=1;i-2*j>=l;j++){
if(s[i]==s[i-j]&&s[i]==s[i-2*j])
l=i-2*j;
}
ans+=l;
}
printf("%lld\n",ans);
return 0;
}