【Week 12 CSP-M3 C】咕咕东学英语

题目描述

输入输出格式

样例

数据规模

思路

因为字符串中只有两种字符，正向考虑子串是否满足要求的判断条件设计比较困难，而反向考虑不满足要求的子串其判断条件设计较为简单，不满则条件的子串为AB…B（n个B，n>=1）,或BA…A（n个A，n>=1）或A…AB，或B…BA只有这四种。只要在所有子串总数中减去这四种类型子串即可，字符数量大于2的子串数量为n*(n-1)/2。
这里采用第一题的办法，把相同字符段缩为一个整型变量，对于第一，二种类型，即从第二个整型变量开始遍历，他们的和即为这两种子串的总数（取前一段的最后一个字符）；第三，四种从第一个整型遍历到倒数第二个整型，他们的和就是这两种类型子串的总数（取后一段的第一个字符）。不过其中AB，BA这种子串被重复计算，需要在发生重复计算时加上。

代码

#include <iostream>
using namespace std;
const int size=3e5+10;
int note[size];
long long  n,count;
long long solve()
{
	long long ans=n*(n-1)/2; 
	for(int i=1;i<count;i++)
		ans-=note[i];
	for(int i=2;i<=count;i++)
	{
		ans-=note[i];
		ans++;
	}
	return ans;
}
int main(int argc, char** argv) {
	scanf("%d",&n);getchar();
	char last,now;
	scanf("%c",&last);
	int num=1;count=0;
	for(int i=1;i<n;i++)
	{
		scanf("%c",&now);
		if(now==last)num++;
		else
		{
			note[++count]=num;
			num=1;
			last=now;
		}
	}
	note[++count]=num;
	long long ans=solve();
	printf("%lld\n",ans);
	return 0;
}