每日一题----2022.12.26_farmer john 最近购入了 n 头新的奶牛(3≤n≤5×10^5),每头奶牛的品种是更赛牛(-CSDN博客

本文链接：https://blog.csdn.net/m0_73882087/article/details/128460363

题目

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

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

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

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

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

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

输入格式

输入的第一行包含 N。

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

输出格式

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

数据范围

3≤N≤5×10^5

样例

题解

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

typedef long long LL;

const int N = 500010;

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

int main()
{
    scanf("%d", &n);
    scanf("%s", str);

    for (int i = 0, h = 0, g = 0; i < n; i ++ )
        if (str[i] == 'G') l[i] = h, h = 0, g ++ ;
        else l[i] = g, g = 0, h ++ ;

    for (int i = n - 1, h = 0, g = 0; i >= 0; i -- )
        if (str[i] == 'G') r[i] = h, h = 0, g ++ ;
        else r[i] = g, g = 0, h ++ ;

    LL res = 0;
    for (int i = 0; i < n; i ++ )
        res += (LL)l[i] * r[i] + max(l[i] - 1, 0) + max(r[i] - 1, 0);

    printf("%lld\n", res);
    return 0;
}

#include<iostream>
#include<string>
using namespace std;
typedef long long ll;
#define N 500010
int l[N], r[N];
int main()
{
	int a; string b;
	cin >> a >> b;
	for (int i = 0, g = 0, h = 0; i < a; i++)
		if (b[i] == 'G')l[i] = h, h = 0, g++;
		else l[i] = g, g = 0, h++;
	for (int i = a - 1, g = 0, h = 0; i >= 0; i--)
		if (b[i] == 'G')r[i] = h, h = 0, g++;
		else r[i] = g, g = 0, h++;
	ll res = 0;
	for (int i = 0; i < a; i++)
		res += (ll)l[i] * r[i] + max(l[i] - 1, 0) + max(r[i] - 1, 0);
	cout << res;
}