【咕咕东学英语】CSP模测T4

题目

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

题目大意

本题题意较难理解。题目中定义了Delicious串：Delicious串中的每一个字符都属于一个大于1的回文子串。实际就是该串中的每一个字符都在该串的一个子串中，且这个子串是一个长度大于一的回文串。题目给出一段只包含A和B的序列问其中有多少个Delicious串。

解题思路

本题为第三周的csp模拟题，虽然一开始读懂了题意，但没有找到做题关键且出现了编译错误，导致本题没有分数。
首先编译错误是出在gets()函数上，这里要格外注意，csp考试中可能同样无法编译通过，以后需要用其他方法来替代。
另外在做题思路上，起初我发现本题可以转化为回文串拼接问题，所以打算先用dp找到所有回文子串的起止位置，然后将可以拼接的组合，最终的结果就是所有回文子串数量和拼接得到的串的数量。但是这种方法对于3×10^5的数据量显然无法保存dp数组，同时时间复杂度也很危险。
最终发现本题可以换一个思路考虑，就是用所有长度大于一的子串数量减去不合规则的子串数量。长度为n的字符串的长度大于一的子串数为n×(n-1)/2，之后寻找所有形如“A…AB…B”和“B…BA…A”的子串，因为这些子串和子串的同样格式的子串显然是不满足题意的，所以最后用子串总数减去这些不合规则的就好。计算推导可以得出cnta个A和cntb个B构成的这种子串的同类型子串个数为(cnta+cntb-1)，遍历字符串减去所有不合规则的就是最终答案。

具体代码

#include<iostream>
#include<cstring>
#include<cstdio>
#include<string>
#include<algorithm>
#include<cmath>
#include<cstdlib>
#include<iomanip>
#define ll long long
#define MAXN 1005 

using namespace std;

int main()
{
	ll n,ans;
	ll cnta = 0,cntb = 0;
	string s;
	char last;
	cin >> n;
	cin >> s;
	ans = n * (n-1) / 2;
	for(int i = 0; i < n; i++)
	{
		if(i == 0)
		{
			last = s[i];
			if(s[i] == 'A') cnta++;
			else cntb++;
			continue;
		}
		if(last != s[i])
		{
			if(cnta && cntb)
			{
				ans -= (cnta + cntb - 1);
				if(s[i] == 'A') cnta = 0;
				else cntb = 0;
			}
			last = s[i];
		}
		if(s[i] == 'A') cnta++;
		else cntb++;
	}
	if(cnta && cntb)
	{
		ans -= (cnta + cntb - 1);
	}
	cout << ans << endl;
	return 0;
}