孤独的照片

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

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

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

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

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

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

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

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

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

数据范围
3≤N≤5×105
输入样例:
5
GHGHG
输出样例:
3
样例解释
这个例子中的每一个长为 3
的子串均恰好包含一头更赛牛或荷斯坦牛——所以这些子串表示孤独的照片,并会被 Farmer John 扔掉。

所有更长的子串(GHGH、HGHG 和 GHGHG)都可以被接受。
`#include<bits/stdc++.h>
using namespace std;
const int N=500010;
int l[N],r[N];

int main()
{
int n;cin>>n;
string s;cin>>s;
for(int i=0,h=0,g=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,h=0,g=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);
}
cout<<res;
return 0;
}`
为什么需要这三部分?
l[i] * r[i]:

这部分计算的是以 s[i] 为中心,左右两边都有不同字符的子序列。

例如,如果 s[i] = 'G',左边有 l[i] 个 H,右边有 r[i] 个 H,那么可以形成 l[i] * r[i] 个符合条件的子序列。

max(0, l[i] - 1):

这部分计算的是以 s[i] 为中心,只有左边有不同字符的子序列。

例如,如果 s[i] = 'G',左边有 l[i] 个 H,那么可以形成 l[i] - 1 个符合条件的子序列(因为子序列长度至少为 3)。

max(0, r[i] - 1):

这部分计算的是以 s[i] 为中心,只有右边有不同字符的子序列。

例如,如果 s[i] = 'G',右边有 r[i] 个 H,那么可以形成 r[i] - 1 个符合条件的子序列(因为子序列长度至少为 3)。

posted @   爱学习的小许?  阅读(3)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 地球OL攻略 —— 某应届生求职总结
· 周边上新:园子的第一款马克杯温暖上架
· Open-Sora 2.0 重磅开源!
· 提示词工程——AI应用必不可少的技术
· .NET周刊【3月第1期 2025-03-02】
点击右上角即可分享
微信分享提示