题目描述
咕咕东很聪明,但他最近不幸被来自宇宙的宇宙射线击中,遭到了降智打击,他的英语水平被归 零了!这一切的始作俑者宇宙狗却毫不知情!
此时咕咕东碰到了一个好心人——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的回文串中
在只有两种字符的时候,我们发现中间的字母不影响,一定属于某个回文串。在两边的字母,按照其相邻的是否与之相同判断。
最后,我们正向反向遍历解决的时候,有重复计算,去掉。
1 #include<iostream>
2 #include <string>
3 #include<cstring>
4 using namespace std;
5
6 long long solve(string str,long long len){
7 long long ans = len*(len-1)/2;
8 long long k=0;
9 //AB型作为右端
10 for(long long i=1;i<len;i++){
11 if(str[i]!=str[i-1]){
12 ans -= (i-k-1);
13 k=i;
14 }
15 }
16 //AB型作为左端
17 k=len-1;
18 for(long long i=len-2;i>=0;i--){
19 if(str[i]!=str[i+1]){
20 ans -= (k-i);
21 k=i;
22 }
23 }
24 return ans;
25 }
26 int main(){
27 long long n;
28 string s;
29 cin>>n;cin>>s;
30
31 cout<<solve(s,n)<<endl;
32 return 0;
33 }