CSP 模拟3——C

咕咕东学英语

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

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

输出描述

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

样例输入

5 
AABBB

样例输出

6

问题分析

感觉最难的就是理解delicious的定义，当且仅当它的每一个字符都属于一个 大于1的回文子串中，注意这里是回文子串。一开始就是没注意他是子串所以走了弯路。
那么不满足此定义的子串有：AAB和BAA两种类型。
对于第一种，从左向右遍历，遇到s[l] != s[l+1]，则不合条件的子串数+1
对于第二种，从右向左遍历，遇到s[r] != s[r-1]，则不合条件的子串数+1
注意两次遍历后要把重合的去掉在求和，即为不合条件的子串数。

总的子串数目：n*(n-1)/2，减去不合条件的即为答案。

代码实现

#include <iostream>
#include <string>
using namespace std;
 
long long int n;
string s;
 
int main() 
{
	cin>>n;
	cin>>s;
	int l,r;
	l=n-2;
	r=1;
	
	long long int childlen = n*(n-1)/2;
	long long int no=0,len=1;
	
	for(;r<n;r++) 
	{
		if(s[r] != s[r-1]) 
		{
			no += len;
			len = 1;
		}
		else len++;
	}
	
    len = 1;
	for(;l>=0;l--) 
	{
		if(s[l] != s[l+1]) 
		{
			no += len-1;
			len = 1;
		}
		else len++;
	}
	cout<<childlen - no<<endl; 
	return 0;
}