Week12 CSP-M3模拟 T4 咕咕东学英语

题目描述

咕咕东很聪明，但他最近不幸被来自宇宙的宇宙射线击中，遭到了降智打击，他的英语水平被归 零了！这一切的始作俑者宇宙狗却毫不知情！
此时咕咕东碰到了一个好心人——TT，TT在吸猫之余教咕咕东学英语。今天TT打算教咕咕东字母A 和字母B，TT给了咕咕东一个只有大写A、B组成的序列，让咕咕东分辨这些字母。
但是咕咕东的其他学科水平都还在，敏锐的咕咕东想出一个问题考考TT：咕咕东问TT这个字符串 有多少个子串是Delicious的。
TT虽然会做这个问题，但是他吸完猫发现辉夜大小姐更新了，不想回答这个问题，并抛给了你， 你能帮他解决这个问题吗？
Delicious定义：对于一个字符串，我们认为它是Delicious的当且仅当它的每一个字符都属于一个 大于1的回文子串中。

样例

Input

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

Output

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

样例输入

5 
AABBB

样例输出

6

思路

要求求出delicious的子串，只需要用所有长度大于2的子串减去不满足delicious的子串即可，可知只有AAAAB、BAAAAA、ABBBBBB、BBBBBA这几种形式不满足delicious，求出这种子串的数量即可

代码

#include <iostream>
using namespace std;

int main()
{
	long long n;
	cin>>n;
	string s;
	cin>>s;
	
	long long sum=0,ans=0;
	sum=n*(n-1)/2;
	int flag=0;
	
	for(int i=0;i<n;i++)
	{
		if(i&&s[i]!=s[i-1])
		{
			ans=ans+(i-flag);
			flag=i;
		}
		else if(i&&flag!=0) ans++;
	}
	cout<<sum-ans<<endl;
}