AB-string CodeForces - 1238D

AB-string CodeForces - 1238D

题意

如果一个字符串的每个字母，属于至少一个（长度大于1）的回文串，则称这个字符串为good。给出仅由AB组成的字符串，求出good子串数量。

思路

由于字符串只能由AB两种字符构成，如若其不为good
串，其结构应为
$$\begin{gathered} AAA...AB \\ BBB...BA \\ AB...BBB \\ BA...AAA \end{gathered}$$
所以可以通过总的子串数$\frac{n(n-1)}{2}$减去不为good串的子串数量。

注意

使用longlong，否则的溢出

AC代码

#include <iostream>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#define ll long long
using namespace std;

int main()
{
	ll n;
	int f=0;
	ll ans=0;
	string s;
	cin>>n>>s;
	ans= n*(n-1)/2;
	for(int i=1;i<n;i++)
	{
		if(s[i]!=s[i-1])
		{
			ans-=i-f;
			f=i;
		}
		else if(f)
			ans--;
	}
	printf("%lld",ans);
	return 0;
}