孤独的照片(c++实现)

题目

---

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

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

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

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

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

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

---

输入

---

输入的第一行包含 N。

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

---

输出

---

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

---

代码

#include<iostream>
using namespace std;
const int N = 5e5+10;
int n;
char s[N];
int l[N],r[N];
int main(){
    scanf("%d",&n);
    
    scanf("%s",s);
    
    for(int i=0,g=0,h=0;i<n;i++){
        if(s[i] == 'G') l[i] = h,h=0,g++;
        else l[i] = g,g=0,h++;
    }
    
    for(int i=n-1,g=0,h=0;i>=0;i--){
        if(s[i] == 'G') r[i] = h,h=0,g++;
        else r[i] = g,g=0,h++;
    }
    
    long long res = 0;
    for(int i=0;i<n;i++) res += (long long)l[i]*r[i] + max(0,l[i]-1)+max(0,r[i]-1);
    printf("%lld",res);
}#include<iostream>
using namespace std;
const int N = 5e5+10;
int n;
char s[N];
int l[N],r[N];
int main(){
    scanf("%d",&n);
    
    scanf("%s",s);
    
    for(int i=0,g=0,h=0;i<n;i++){
        if(s[i] == 'G') l[i] = h,h=0,g++;
        else l[i] = g,g=0,h++;
    }
    
    for(int i=n-1,g=0,h=0;i>=0;i--){
        if(s[i] == 'G') r[i] = h,h=0,g++;
        else r[i] = g,g=0,h++;
    }
    
    long long res = 0;
    for(int i=0;i<n;i++) res += (long long)l[i]*r[i] + max(0,l[i]-1)+max(0,r[i]-1);
    printf("%lld",res);
}