acwing 4261孤独的照片

题目

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

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

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

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

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

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

思路

1、左边不同牛的数量 * 右边不同牛的数量 + 左边不同牛的数量（减去只有一头牛的情况） + 右边不同牛的数量（同上）

代码

#include<bits/stdc++.h>
using namespace std;
const int N = 500050;
int l[N],r[N];

int main(){
    int n; cin >> n;
    string s; cin >> s;
    int g = 0, h  = 0;
    for(int i = 0; i < n; i++){
        if(s[i] == 'G'){
            l[i] = h, h = 0,g++;
        }else{
            l[i] = g, g = 0,h++;
        }
    }
    g = 0, h = 0;
    for(int i = n - 1; i >= 0; i--){
        if(s[i] == 'G'){
            r[i] = h, h = 0, g++;
        }else{
            r[i] = g, g = 0, h++;
        }
    }
    long long ans = 0;
    for(int i = 0; i < n; i++){
        if(l[i] + r[i] > 1){
            
            ans += (long long)l[i] * r[i] + max(l[i] - 1, 0) + max(r[i] - 1, 0);
        }
    }
    cout << ans;
    return 0;
}