AB-string CodeForces - 1238D

CodeForces-1238D题解

最新推荐文章于 2022-04-17 15:50:41 发布

原创最新推荐文章于 2022-04-17 15:50:41 发布 · 159 阅读

0 ·

CC 4.0 BY-SA版权

acm训练专栏收录该内容

10 篇文章

订阅专栏

本文解析了CodeForces-1238D题目——AB-string问题，该题要求找出由AB组成的字符串中所有好的子串数量。通过对字符串结构的观察，提出了一种快速计算的方法，并给出了完整的AC代码。

AB-string CodeForces - 1238D

题意

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

思路

由于字符串只能由AB两种字符构成，如若其不为good
串，其结构应为
$\\ BBB...BA\\ AB...BBB\\ BA...AAA\\$
所以可以通过总的子串数 $n(n−1)2\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;
}