AB-string CodeForces - 1238D
题意
如果一个字符串的每个字母,属于至少一个(长度大于1)的回文串,则称这个字符串为good。给出仅由AB组成的字符串,求出good子串数量。
思路
由于字符串只能由AB两种字符构成,如若其不为good
串,其结构应为
A
A
A
.
.
.
A
B
B
B
B
.
.
.
B
A
A
B
.
.
.
B
B
B
B
A
.
.
.
A
A
A
AAA...AB \\ BBB...BA\\ AB...BBB\\ BA...AAA\\
AAA...ABBBB...BAAB...BBBBA...AAA
所以可以通过总的子串数
n
(
n
−
1
)
2
\frac{n(n-1)}{2}
2n(n−1)减去不为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;
}