思路:
算出所有长度大于2的子字符串。遍历整个字符串记录字符AB交换的位置,通过一定的规律,遇到字母不同的就计算个数。所以结果即为总的字符串数量减去不符合的字符串的数量。
代码:
#include<iostream>
#include<cstring>
using namespace std;
int main()
{
long long n;
cin>>n;
string s;
cin>>s;
long long cnp=1;
int flag=0;
long long ans=0;
for(int i=1;i<n;i++)
{
if(s[i]!=s[i-1])
{
ans+=cnp;
cnp=0;
flag=1;
}
else if(flag)
{
ans++;
}
cnp++;
}
ans=(long long)(n*(n-1)/2)-ans;
cout<<ans;
return 0;
}