孤独的照片

Farmer John 最近购入了 N头新的奶牛，每头奶牛的品种是更赛牛（Guernsey）或荷斯坦牛（Holstein）之一。

奶牛目前排成一排，Farmer John 想要为每个连续不少于三头奶牛的序列拍摄一张照片。

然而，他不想拍摄这样的照片，其中只有一头牛的品种是更赛牛，或者只有一头牛的品种是荷斯坦牛——他认为这头奇特的牛会感到孤立和不自然。

在为每个连续不少于三头奶牛的序列拍摄了一张照片后，他把所有「孤独的」照片，即其中只有一头更赛牛或荷斯坦奶牛的照片，都扔掉了。

给定奶牛的排列方式，请帮助 Farmer John 求出他会扔掉多少张孤独的照片。

如果两张照片以不同位置的奶牛开始或结束，则认为它们是不同的。

输入格式

输入的第一行包含 N。

输入的第二行包含一个长为 N的字符串。如果队伍中的第 i 头奶牛是更赛牛，则字符串的第 i 个字符为 G。否则，第 i头奶牛是荷斯坦牛，该字符为 H。

输出格式

输出 Farmer John 会扔掉的孤独的照片数量。

数据范围

3≤N≤5×10^5

输入样例：

5
GHGHG

输出样例：

3

样例解释

这个例子中的每一个长为 3的子串均恰好包含一头更赛牛或荷斯坦牛——所以这些子串表示孤独的照片，并会被 Farmer John 扔掉。

所有更长的子串（GHGH、HGHG 和 GHGHG）都可以被接受。

题解：
（1）暴力

这还是比较直观的吧

在为每个连续不少于三头奶牛的序列拍摄了一张照片后，他把所有「孤独的」照片，即其中只有一头更赛牛或荷斯坦奶牛的照片，都扔掉了。

设更赛牛是g，荷斯坦奶牛是s，由上面的信息可知，g+s≥3，且(g=1或者s=1)。

bool check(int g,int h)
{
    if(g+h>=3&&(g==1||h==1)) return true;
    return false;
}

我们可以先固定左边界，然后再不断枚举右边界，看看是不是满足条件，满足条件答案加1。

代码如下：

#include <iostream>
#include<string>
using namespace std; 

bool check(int g,int h)
{
    if(g+h>=3&&(g==1||h==1)) return true;
    return false;
}

int n,g,h; 
string s;

int main(){ 
    cin>>n;
    cin>>s;
    
    long long res=0;
    for(int i=0;i<s.size();i++)
    { 
        g=0,h=0;//每次都要把他们置0 
        for(int j=i;j<s.size();j++)
        { 
            if(s[j]=='G') g++; 
            else h++; 
            if(check(g,h)) res++; 
        } 
    }
    cout<<res<<endl; 
    return 0; 
}

（2）因为数据过大，所以可能会导致超时，现在我们来用一种新的方法来解决问题，也就是贡献法。

经过上面的分析，相信你肯定知道了我们要统计有字母G的左边多少个连续的H，右边有多少个连续的H，还有就是字母H的左边多少个连续的G，右边有多少个连续的G。那怎么来实现这个想法呢。很简单吧，先看左边吧，右边同理的，使用for循环遍历整个字符串，分别统计连续的G和H的个数sg,sh。假如如果遇到了G，那就这个位置的值就等于之前有多少个连续的H个个数。然后既然这个字母是G，那我们就可以继续统计G的个数了，也就是sg++呗。那之前统计的sh就等于0了，为什么呢，因为到G了啊，H已经不连续了，所以要让他等于0了。假如如果遇到了H，那就这个位置的值就等于之前有多少个连续的G个个数。然后既然这个字母是H，那我们就可以继续统计H的个数了，也就是sh++呗，那之前统计的sg就等于0了。

代码如下：

#include<iostream>
using namespace std;

const int N=5*1e5+10;
char a[N];
int l[N],r[N];
int n;

int main()
{
    cin>>n;
    for(int i=0;i<n;i++) cin>>a[i];
    
    //左边
    for(int i=0,sg=0,sh=0;i<n;i++)
    {
        if(a[i]=='G') l[i]=sh,sg++,sh=0;
        else l[i]=sg,sh++,sg=0;
    }
    
    //右边
    for(int i=n-1,sg=0,sh=0;i>=0;i--)
    {
        if(a[i]=='G') r[i]=sh,sg++,sh=0;
        else r[i]=sg,sh++,sg=0;
    }
    long long res=0;
    for(int i=0;i<n;i++)
    {
        res+=(long long)l[i]*r[i]+max(l[i]-1,0)+max(r[i]-1,0);
    }
    cout<<res<<endl;
    return 0;
}