题目
题目描述
咕咕东很聪明,但他最近不幸被来自宇宙的宇宙射线击中,遭到了降智打击,他的英语水平被归 零了!这一切的始作俑者宇宙狗却毫不知情!
此时咕咕东碰到了一个好心人——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的回文子串。实际就是该串中的每一个字符都在该串的一个子串中,且这个子串是一个长度大于一的回文串。题目给出一段只包含A和B的序列问其中有多少个Delicious串。
解题思路
本题为第三周的csp模拟题,虽然一开始读懂了题意,但没有找到做题关键且出现了编译错误,导致本题没有分数。
首先编译错误是出在gets()函数上,这里要格外注意,csp考试中可能同样无法编译通过,以后需要用其他方法来替代。
另外在做题思路上,起初我发现本题可以转化为回文串拼接问题,所以打算先用dp找到所有回文子串的起止位置,然后将可以拼接的组合,最终的结果就是所有回文子串数量和拼接得到的串的数量。但是这种方法对于3×10^5的数据量显然无法保存dp数组,同时时间复杂度也很危险。
最终发现本题可以换一个思路考虑,就是用所有长度大于一的子串数量减去不合规则的子串数量。长度为n的字符串的长度大于一的子串数为n×(n-1)/2,之后寻找所有形如“A…AB…B”和“B…BA…A”的子串,因为这些子串和子串的同样格式的子串显然是不满足题意的,所以最后用子串总数减去这些不合规则的就好。计算推导可以得出cnta个A和cntb个B构成的这种子串的同类型子串个数为(cnta+cntb-1),遍历字符串减去所有不合规则的就是最终答案。
具体代码
#include<iostream>
#include<cstring>
#include<cstdio>
#include<string>
#include<algorithm>
#include<cmath>
#include<cstdlib>
#include<iomanip>
#define ll long long
#define MAXN 1005
using namespace std;
int main()
{
ll n,ans;
ll cnta = 0,cntb = 0;
string s;
char last;
cin >> n;
cin >> s;
ans = n * (n-1) / 2;
for(int i = 0; i < n; i++)
{
if(i == 0)
{
last = s[i];
if(s[i] == 'A') cnta++;
else cntb++;
continue;
}
if(last != s[i])
{
if(cnta && cntb)
{
ans -= (cnta + cntb - 1);
if(s[i] == 'A') cnta = 0;
else cntb = 0;
}
last = s[i];
}
if(s[i] == 'A') cnta++;
else cntb++;
}
if(cnta && cntb)
{
ans -= (cnta + cntb - 1);
}
cout << ans << endl;
return 0;
}