CodeForces - 1307C Cow and Message(贡献计算)

💜 💜 💜
被题目要求的等差数列吓到了,其实由这个要求可以发现子串长度大于3是没有贡献的,因为只要出现长度为3的串就一定存在长为2的串。

所以我们对长为2和长为1的串分别计算一下贡献即可,计算长为2的串的时候枚举当前字符作为第二个字母的贡献,最后加和

#define int ll
string s;
int dp[26][26],num[26];
void solve()
{
    cin>>s;
    int n=sz(s);
    for(int i=0;i<n;++i)
    {
        
        rep(j,26) 
        {
            dp[j][s[i]-'a']+=num[j];
        }
        ++num[s[i]-'a'];
    }
    int ans=0;
    rep(i,26) ans=max(ans,num[i]);
    rep(i,26) rep(j,26) ans=max(ans,dp[i][j]);
    cout<<ans<<endl;
}
signed main()
{
    solve();
    return 0;
}