题目描述
咕咕东很聪明,但他最近不幸被来自宇宙的宇宙射线击中,遭到了降智打击,他的英语水平被归 零了!这一切的始作俑者宇宙狗却毫不知情! 此时咕咕东碰到了一个好心人——TT,TT在吸猫之余教咕咕东学英语。今天TT打算教咕咕东字母A 和字母B,TT给了咕咕东一个只有大写A、B组成的序列,让咕咕东分辨这些字母。 但是咕咕东的其他学科水平都还在,敏锐的咕咕东想出一个问题考考TT:咕咕东问TT这个字符串 有多少个子串是Delicious的。 TT虽然会做这个问题,但是他吸完猫发现辉夜大小姐更新了,不想回答这个问题,并抛给了你, 你能帮他解决这个问题吗? Delicious定义:对于一个字符串,我们认为它是Delicious的当且仅当它的每一个字符都属于一个 大于1的回文子串中。
输入描述
输入第一行一个正整数n,表示字符串长度 接下来一行,一个长度为n只由大写字母A、B构成的字符串。
输出描述
输出仅一行,表示符合题目要求的子串的个数。
样例输入
5
AABBB
样例输出
6
样例解释
对于该样例,符合条件的六个子串分别是:
s1-s2,s1-s4,s3-s4,s3-s5,s4-s5
数据组成
思路:
将相同的字母分成一段,发现子串只要段数大于3就一定符合条件,分析段数为1和2的情况,只有A…AB、BA…A、B…BA、AB…B不符合情况。
综上符合条件的子串情况复杂,而不符合的情况较简单,正难则反,归结于求每一段的元素数目,注意防止AB、BA前后重复计算。
代码:
#include <iostream>
using namespace std;//别被回文 迷倒了,主要只有AB两个字母才是关键
const int Max_size = 3e5+10;
struct Point
{
int num = 0;
int value;
}p[Max_size];
int main()
{
string str;
int n;
cin >> n;
cin >> str;
int k = 0;
p[k].value = str[0];
p[k].num++;
int last_value = str[0];
for(int i = 1; i < n; i++)
{
if(str[i] == last_value)
{
p[k].num++;
}
else{
k++;
p[k].num++;
p[k].value = str[i];
last_value = str[i];
}
}
long long ans = (long long)n*(n-1)/2;//总可能的序列数
for(int i = 0; i < k; i++)
{
ans -= p[i].num;
ans -= p[i+1].num;
ans++;// AAAABB AB算了两次
}
cout << ans;
return 0;
}
总结:
1.做题过于死板,看到回文就先入为主,思考判断回文的方法,如果所有子串的每个字母都分析一遍,不但复杂度显然太大,而且还被迷住,不会做了。此题只有A、B两种字母,两种情况,首先应该代入题目,寻找符合这个题的方法。
2.本质上还是求每个段的元素数目,用一个结构体存储每个段的value,以及元素数目,last_value记录前驱,遍历全部元素就可以了