孤独的照片（2023寒假每日一题 2）

Farmer John 最近购入了 $N$

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

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

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

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

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

输入格式
输入的第一行包含 $N$。

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

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

数据范围
$3≤N≤5×10^5$
输入样例：

5
GHGHG

输出样例：

3

样例解释
这个例子中的每一个长为 $3$

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

---

枚举以每个位置的字符为单字符的孤独的照片的字符串
而假如此位置是 ​​​H​​​，将紧挨着此位置左边连续的 ​​G​​​ 的数量 计作 L，此位置右边连续的 ​​G​​​ 的数量 计作 R
以此位置为孤独的照片的字符串的数量为 L * R + (L - 1) + (R - 1)

#include<iostream>

using namespace std;

typedef long long LL;

const int N = 500010;

int n;
int l[N], r[N];
char str[N];

int main(){
    
    scanf("%d%s", &n, str);
    
    for(int i = 0, g = 0, h = 0; i < n; i++)
        if(str[i] == 'G') l[i] = h, g++, h = 0;
        else l[i] = g, h++, g = 0;
    
    for(int i = n - 1, g = 0, h = 0; i >= 0; i--)
        if(str[i] == 'G') r[i] = h, g++, h = 0;
        else r[i] = g, h++, g = 0;
        
    LL res = 0;
    for(int i = 0; i < n; i++)
        res += (LL)l[i] * r[i] + max(0, l[i] - 1) + max(0, r[i] - 1);
    
    printf("%lld\n", res);
    
    return 0;
}