week12 CSP模拟T4咕咕东学英语

T4 咕咕东学英语

求一个只有大写A、B组成的字符串有多少个子串(由原字符串中位置连续的字符构成的字符串)是Delicious的。
Delicious定义：对于一个字符串，我们认为它是Delicious的当且仅当它的每一个字符都至少属于一个长度大于1且被完整包含在这个字符串中的回文子串中。
某个字符串是回文的，即这个字符串从左往右读和从右往左读是相等的。
子串(substring)的概念：比如 abc 是 ddddabcdddd 的子串。

Input

输入第一行一个正整数n，表示字符串长度
接下来一行，一个长度为n只由大写字母A、B构成的字符串。

Output

输出仅一行，表示符合题目要求的子串的个数。

Sample Input

5
AABBB

Sample Output

6

我的思路：
这道题是要求满足Delicious连续子串的个数，即其中的所有字符都在一个长度至少为2的回文子串中。因为字符串中只有A、B两种字符，所以不符合Delicious的子串的情况只会有A…AB、B…BA、BA…A、AB…B。所以只要把这些情况在所有子串（长度为n的字符串，连续子串的个数为n*(n-1)/2）中去除，就可以得到最后的结果。（这里在判断的时候，可以从前向后判断A…AB、B…BA，从后向前判断BA…A、AB…B，注意这里在每次满足判断条件时，会多减去一个AB或BA，要把多减去的加上）

我的总结：
这道题，我最初的想法是遍历字符串，然后写一个判断是否满足要求的函数，因为涉及到子串两端的变化，时间复杂度是很高的。但是，如果先把不满足的情况列出来，然后再进行查找遍历，会简单高效许多。

我的代码：

#include<iostream>
#include<cstring>
using namespace std;

long long cur,n,ans;
string st;

int main()
{
	cin>>n;
	cin>>st;
	ans = (n*(n-1))/2;
	cur=0;
	for(int i=1;i<n;i++)
	{
		if(st[i]!=st[i-1])
		{
			ans -= (i-cur); 
			ans++;
			cur=i;
		}
	}
	cur=n-1;
	for(int i=n-2;i>=0;i--)
	{
		if(st[i]!=st[i+1])
		{
			ans -= (cur-i);
			cur=i;
		}
	}
	cout<<ans;
	return 0;
 }