Farmer John 最近购入了 N头新的奶牛,每头奶牛的品种是更赛牛(Guernsey)或荷斯坦牛(Holstein)之一。
奶牛目前排成一排,Farmer John 想要为每个连续不少于三头奶牛的序列拍摄一张照片。
然而,他不想拍摄这样的照片,其中只有一头牛的品种是更赛牛,或者只有一头牛的品种是荷斯坦牛——他认为这头奇特的牛会感到孤立和不自然。
在为每个连续不少于三头奶牛的序列拍摄了一张照片后,他把所有「孤独的」照片,即其中只有一头更赛牛或荷斯坦奶牛的照片,都扔掉了。
给定奶牛的排列方式,请帮助 Farmer John 求出他会扔掉多少张孤独的照片。
如果两张照片以不同位置的奶牛开始或结束,则认为它们是不同的。
输入格式
输入的第一行包含 N
。
输入的第二行包含一个长为 N的字符串。如果队伍中的第 i头奶牛是更赛牛,则字符串的第 i个字符为 G。否则,第 i 头奶牛是荷斯坦牛,该字符为 H。
输出格式
输出 Farmer John 会扔掉的孤独的照片数量。
数据范围
3≤N≤5×105
输入样例:
5
GHGHG
输出样例:
3
样例解释
这个例子中的每一个长为 3
的子串均恰好包含一头更赛牛或荷斯坦牛——所以这些子串表示孤独的照片,并会被 Farmer John 扔掉。
所有更长的子串(GHGH、HGHG 和 GHGHG)都可以被接受。
这里我们只需要用一个string将传入的字符串给读取,然后我们挨着遍历字符串中的每一个元素,也就是遍历每一头奶牛。然后寻找每一头奶牛左侧有多少头连续的跟当前这头奶牛不同品种的奶牛l1,以及这头奶牛右边有多少头连续的跟当前这头奶牛不同品种的奶牛r1;
然后孤独的照片只有三种情况:
①以当前这头奶牛为起点,向左拍摄,有l1-1张孤独的照片
②以当前这头奶牛为起点,向右拍摄,有r1-1张孤独的照片
③以当前的这头奶牛位照片中中间的奶牛,左边有l1种不同的选择,右边有r1种不同的选择,有l1*r1种不同的孤独的照片。
#include<iostream>
#include<string>
#include<algorithm>
typedef long long ll;
using namespace std;
int main()
{
ll N,count=0;
cin>>N;
string s;
cin>>s;
for(ll i=0;i<s.size();i++)
{
ll l=i-1,r=i+1,l1=0,r1=0;
while(l>=0&&s[l]!=s[i]) l--,l1++;
while(r<s.size()&&s[i]!=s[r]) r++,r1++;
count+=l1*r1+max((ll)0,l1-1)+max((ll)0,r1-1);
}
cout<<count<<endl;
}