一、题意
1.简述
咕咕东很聪明,但他最近不幸被来自宇宙的宇宙射线击中,遭到了降智打击,他的英语水平被归
零了!这一切的始作俑者宇宙狗却毫不知情!
此时咕咕东碰到了一个好心人——TT,TT在吸猫之余教咕咕东学英语。今天TT打算教咕咕东字母A
和字母B,TT给了咕咕东一个只有大写A、B组成的序列,让咕咕东分辨这些字母。
但是咕咕东的其他学科水平都还在,敏锐的咕咕东想出一个问题考考TT:咕咕东问TT这个字符串
有多少个子串是Delicious的。
TT虽然会做这个问题,但是他吸完猫发现辉夜大小姐更新了,不想回答这个问题,并抛给了你,
你能帮他解决这个问题吗?
Delicious定义:对于一个字符串,我们认为它是Delicious的当且仅当它的每一个字符都属于一个
大于1的回文子串中。
2.输入格式
输入第一行一个正整数n,表示字符串长度
接下来一行,一个长度为n只由大写字母A、B构成的字符串。
3.输出格式
输出仅一行,表示符合题目要求的子串的个数。
4.样例
Input
5
AABBB
Output
6
Hint
对于该样例,符合条件的六个子串分别是:
二、算法
主要思路
该题如果从正面想,找有多少种正确的格式,很难找,但是从反面想,找有多少种错误的格式,就比较有规律了。可以写写看看,错误的格式只有两种:
AB、ABB、ABBB…
BA、BAA、BAAA…
AB、AAB、AAAB…
BA、BBA、BBBA…
所以我们可以采用总的子串的个数减去错误格式子串的个数来求正确子串的个数。我们可以从头到尾遍历一遍找出前两种格式的子串的个数,之后再从尾到头遍历一遍,找出后两种个格式的子串的个数,需要注意的是AB和BA会被计数两次,所以最后要减去AB和BA子串的个数,就能求出所有错误子串的个数。
而总的子串的个数为n*(n-1))/2
。用总的减去错误的就是正确的。
三、代码
#include <iostream>
#include <vector>
#include <queue>
#include <string>
#include <cstring>
#include <cstdio>
using namespace std;
const int maxn = 3e5+10;
char s_c_str[maxn];
long long int sum = 0;
int main() {
int n;
scanf("%d",&n);
string s;
cin>>s;
strcpy(s_c_str,s.c_str());
int cnt = 0;
int cnt_ab = 0;
char temp = s_c_str[0];
bool b = false;
for (int i = 1; i < n; ++i) {
if (s_c_str[i] != temp){
cnt++;
if (!b) cnt_ab++;
b = true;
}
else if (b){
temp = s_c_str[i-1];
cnt++;
cnt_ab++;
}
}
b = false;
temp = s_c_str[n-1];
for (int i = n-2; i >= 0; --i) {
if (s_c_str[i] != temp){
cnt++;
b = true;
}
else if (b){
temp = s_c_str[i+1];
cnt++;
}
}
sum = ((long long)n*(n-1))/2;
//printf("SUM1:%lld\n",sum);
sum -= cnt - cnt_ab;
// printf("cnt_ab:%d\n",cnt_ab);
//printf("cnt:%d\n",cnt);
printf("%lld",sum);
return 0;
}
/*
*/