所有不合法的只有这些可能:
ABBBBBB
BAAAAAA
AAAAAAB
BBBBBBA
所以把这些去了就可以了。
#include <bits/stdc++.h> #define inf 2333333333333333 #define N 1000010 #define p(a) putchar(a) #define For(i,a,b) for(long long i=a;i<=b;++i) using namespace std; long long n,ans,l,r; char a[N]; void in(long long &x){ long long y=1;char c=getchar();x=0; while(c<'0'||c>'9'){if(c=='-')y=-1;c=getchar();} while(c<='9'&&c>='0'){ x=(x<<1)+(x<<3)+c-'0';c=getchar();} x*=y; } void o(long long x){ if(x<0){p('-');x=-x;} if(x>9)o(x/10); p(x%10+'0'); } signed main(){ in(n); scanf("%s",(a+1)); ans=n*(n-1)/2,l=1; For(i,2,n){ if(a[i]!=a[i-1]){ ans-=i-l-1; l=i; } } r=n; for(int i=n-1;i;i--){ if(a[i]!=a[i+1]){ ans-=r-i; r=i; } } o(ans); return 0; }