第五周模拟测试-3

题目：
描述

咕咕东很聪明，但他最近不幸被来自宇宙的宇宙射线击中，遭到了降智打击，他的英语水平被归零了！这一切的始作俑者宇宙狗却毫不知情！

此时咕咕东碰到了一个好心人——TT，TT在吸猫之余教咕咕东学英语。今天TT打算教咕咕东字母A和字母B，TT给了咕咕东一个只有大写A、B组成的序列，让咕咕东分辨这些字母。

但是咕咕东的其他学科水平都还在，敏锐的咕咕东想出一个问题考考TT：咕咕东问TT这个字符串有多少个子串(由原字符串中位置连续的字符构成的字符串)是Delicious的。

TT虽然会做这个问题，但是他吸完猫发现辉夜大小姐更新了，不想回答这个问题，并抛给了你，你能帮他解决这个问题吗？

Delicious定义：对于一个字符串，我们认为它是Delicious的当且仅当它的每一个字符都至少属于一个长度大于1且被完整包含在这个字符串中的回文子串中。

我们说某个字符串是回文的，即这个字符串从左往右读和从右往左读是相等的。

子串(substring)的概念：比如 abc 是 ddddabcdddd 的子串。

下面是一些Delicious字符串的的例子:

t=AABBBt=AABBB (字符t1t1,t2t2属于回文串t1…t2t1…t2,字符t3t3,t4t4,t5t5属于回文串t3…t5t3…t5)

t=ABAAt=ABAA (字符t1t1,t2t2,t3t3属于回文串t1…t3t1…t3,字符t4t4属于回文串t3…t4t3…t4)

t=AAAAAt=AAAAA (所有字符均属于回文串t1…t5t1…t5)

输入
输入第一行一个正整数n，表示字符串长度

接下来一行，一个长度为n只由大写字母A、B构成的字符串。

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

输入样例 1

5
AABBB
输出样例 1

6
提示

对于该样例，符合条件的六个子串分别是：

s1~s2、s1-s4、s1-s5、s3-s4、s3-s5、s4-s5

即 AA、AABB、AABBB、BB、BBB、BB

| 数据点 | n |

| 1,2 | 10 |

| 3,4 | 100 |

| 5,6 | 233 |

| 7,8,9,10 | 3 × 10^5 |
题目分析：
本题非常绕，读起来很复杂，但是稍加分析就能发现满足的情况：
1、对于同一个字母里面相同的数，可以直接用全排列的方式求出对应的回文子串，当然前提是长度必须大于1，这样我们就可以套用公式来做。

	for(int i=1;i<=count;i++)
	{
		if(b[i]>1)
		{
			ans=ans+(b[i]*(b[i]-1))/2;
		}
	}

当然，如何求出字母出现的次数，就需要和上次作业一样的思路，从前往后一个一个统计，开一个新数组，遇到一样的就个数++，遇到不一样的就找下一个。

	int count=1;
	b[1]=1;
	for(int i=2;i<=n;i++)
	{
		if(a[i]==a[i-1])
		{
			b[count]++;
		}
		else
		{
			b[count]++;
			count++;
		}
	}

2、对于相邻两个不同字母之间组成的回文串来说也一样，首先不能出现一次，否则不成立，然后就可以是上面样例的组合，也就是一个字母能出现几次（长度大于2），另一个字母出现几次（长度大于2），之后相乘就好了。

	for(int i=2;i<=count;i++)
	{
		if(b[i]>1&&b[i-1]>1)
		{
			ans=ans+(b[i-1]-1)*(b[i]-1);
		}
	}

3、对于三个以上的字母也是同理，只不过允许出现一次了，例如ABA。

	for(int i=count;i>=3;i--)
	{
		c[i]=c[i+1]+b[i];
	}
	for(int i=1;i<=count-2;i++)
	{
		ans=ans+b[i]*c[i+2];
	}	

代码如下：

#include<iostream>
using namespace std;
char a[300010];
int b[300010]={0};
int c[300010]={0};
int main()
{
	int n;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
	}
	int count=1;
	b[1]=1;
	for(int i=2;i<=n;i++)
	{
		if(a[i]==a[i-1])
		{
			b[count]++;
		}
		else
		{
			b[count]++;
			count++;
		}
	}
	for(int i=count;i>=3;i--)
	{
		c[i]=c[i+1]+b[i];
	}
	long long ans=0;
	for(int i=1;i<=count;i++)
	{
		if(b[i]>1)
		{
			ans=ans+(b[i]*(b[i]-1))/2;
		}
	}
	for(int i=2;i<=count;i++)
	{
		if(b[i]>1&&b[i-1]>1)
		{
			ans=ans+(b[i-1]-1)*(b[i]-1);
		}
	}
	for(int i=1;i<=count-2;i++)
	{
		ans=ans+b[i]*c[i+2];
	}
	cout<<ans;
}