D. AB-string

D. AB-string

theme:给定一个长度为n的，仅由A、B组成的字符串s，定义一个串t是好的当且仅当t的每一个字符的至少属于一个t的长度>1的回文子串，问s有多少个好的子串？

solution：可发现只有这两种情况的串不是好的即：ABBB、BBBA，即一个A/B + 多个另一个字符。总的子串为n*(n-1)/2 (除去长度为1的)。对于AABBB，不满足条件的子串个数就是2+3-1,即连续A的个数+连续B个数-1。所以遍历一遍S预处理出每个字符连续出现的个数即可。

#include<bits/stdc++.h>
#include<vector>
#include<algorithm>
#define far(i,t,n) for(int i=t;i<n;++i)
#define pb(a) push_back(a)
#define lowbit(x) x&(-x)
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
int inf=0x3f3f3f3f;
const int maxn=300010;

char a[maxn];
vector<int>v;

int main()
{
    int n;
    cin>>n;
    cin>>a;
    int i=0;
    while(i<n)
    {
        char now=a[i];
        int cnt=1;
        ++i;
        while(i<n&&a[i]==now)
        {
            ++cnt;
            ++i;
        }
        v.pb(cnt);
    }
    int sz=v.size();
    ll ans=(ll)n*(n-1)/2;
    far(i,1,sz)
        ans-=v[i]+v[i-1]-1;
    cout<<ans<<endl;
}