E - Good Triple(思维求贡献）

Posted 2022-12-01 CCSU_Cola

题目链接

题意：给一个只包含0和1的字符串S，问在字符串中有多少个区间[L,R]包含S[l]==s[l+k]==s[l+2*k],且L<=l,l+2*k<=R。

思路：因为只包含0和1两种字符，可以知道不构成题目条件的串的长度不会大于9，也就是说只需要暴力求解即可，枚举以L为起点，得到满足条件的最小R。答案加上strlen(s)-R+1即可。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=300010;
char a[maxn];
int main()
    scanf("%s",a+1);
    int k=strlen(a+1);
    ll r=k+2,ans=0;
    for(int i=k-1;i>=1;i--)
        for(int j=1;i+j*2<=k;j++)
            if(a[i]==a[i+j]&&a[i]==a[i+2*j])
                r=min(r,(1ll)*i+j*2);//找到最小的R
                break;
            
        
        ans+=max(k-r+1,0ll);
    
    printf("%lld\\n",ans);