咕咕东学英语

对于这个题，本来我是想通过求出所有的回文序列区间，然后去用现有的区间去拼凑更大的满足题目要求的区间，但是这样做的话难免会遗漏掉某些区间，并且在求回文序列区间时的复杂度最坏可以到达$O(n^3)$，根据数据范围肯定会T。所以可以换一种思路，首先可以求出所有的区间个数，然后通过遍历，将不符合要求的区间删除，时间复杂度为O(n)。

问题描述

今天TT打算教咕咕东字母A和字母B，TT给了咕咕东一个只有大写A、B组成的序列，让咕咕东分辨这些字母。但是咕咕东的其他学科水平都还在，敏锐的咕咕东想出一个问题考考TT：咕咕东问TT这个字符串有多少个子串是Delicious的。 Delicious定义：对于一个字符串，我们认为它是Delicious的当且仅当它的每一个字符都属于一个大于1的回文子串中。

input：
输入第一行一个正整数n，表示字符串长度
接下来一行，一个长度为n只由大写字母A、B构成的字符串。
output：
输出仅一行，表示符合题目要求的子串的个数。

样例输入：

5
AABBB

样例输出：

6

解题思路

对于一个长度为n的字符串，首先我们可以得到所有长度大于2的区间个数，即我们从n个数字中任意取两个数都可以当做一个区间的两个端点，并且长度大于2，所以所以区间个数就是对n个元素取两个元素的排列组合，即(n*(n-1))/2个区间。然后对n个元素进行扫描，只有AAA…B，BBB…A，ABBB…B和BAAA…A这样的区间才无效，所以我们对长度为n的字符串进行正反两次遍历就可以了，由于在两次遍历中，像AB和BA这样的区间被多减去的一次，所以还要加上，最后输出结果。

代码

#include<iostream>
#include<string>
using namespace std;
int main()
{
	long long int n,alln;
	string chuan;
	cin>>n;
	cin>>chuan;
	alln=(n*(n-1))/2;//所以区间个数
	int flag=0,temp=0;
	for(int i=1;i<n;i++)
	{
		if(chuan[i]!=chuan[i-1])//遇到了一个不符合要求的区间，将不符合条件的区间减去
		{
			alln-=(i-flag);
			flag=i;
			temp++;
		} 
	} 
	alln+=temp;//由于反向遍历还会减去重复的，所以要加上
	flag=n-1;
	for(int i=n-2;i>=0;i--)
	{
		if(chuan[i+1]!=chuan[i])
		{
			alln-=(flag-i);
			flag=i;
		}
	}
	cout<<alln<<endl;
}