4261. 孤独的照片(题目1)
Farmer John 最近购入了 N 头新的奶牛,每头奶牛的品种是更赛牛(Guernsey)或荷斯坦牛(Holstein)之一。
奶牛目前排成一排,Farmer John 想要为每个连续不少于三头奶牛的序列拍摄一张照片。
然而,他不想拍摄这样的照片,其中只有一头牛的品种是更赛牛,或者只有一头牛的品种是荷斯坦牛——他认为这头奇特的牛会感到孤立和不自然。
在为每个连续不少于三头奶牛的序列拍摄了一张照片后,他把所有「孤独的」照片,即其中只有一头更赛牛或荷斯坦奶牛的照片,都扔掉了。
给定奶牛的排列方式,请帮助 Farmer John 求出他会扔掉多少张孤独的照片。
如果两张照片以不同位置的奶牛开始或结束,则认为它们是不同的。
输入格式
输入的第一行包含 N。
输入的第二行包含一个长为 N 的字符串。如果队伍中的第 i 头奶牛是更赛牛,则字符串的第 i 个字符为
G
。否则,第 i 头奶牛是荷斯坦牛,该字符为H
。
输出格式
输出 Farmer John 会扔掉的孤独的照片数量。
数据范围
3≤N≤5×1053≤N≤5×105
输入样例:
5
GHGHG
输出样例:
3
样例解释
这个例子中的每一个长为 3 的子串均恰好包含一头更赛牛或荷斯坦牛——所以这些子串表示孤独的照片,并会被 Farmer John 扔掉。
所有更长的子串(
GHGH
、HGHG
和GHGHG
)都可以被接受。
题解
思路:
大致可分为三种情况(以H为例,G是同理的):
如果输入的字符串是:GGGHGG
1.H在中间时
有这几种情况:
左边1个G时:GHG,GHGG
左边2个G时:GGHG,GGHGG
左边3个G时:GGGHG,GGGHGG
总共6种
根据排列组合,我们可以发现,左边连续G的个数*右边连续G的个数=H在中间时候的可能性
2.H在左边时
有这1种情况:HGG(即右边连续个数-1=情况数)
3.H在右边时
有这2种情况:GGH,GGGH(即左边连续个数-1=情况数)
最后,把他们三者的种类数相加即得最后结果
输入的字符串如果是:GGGHGGHGGG
你刚刚举的是一个H,要是输入2个H怎么办呢?
其实是同样道理,以H为单位分成两部分,分别以上面方法讨论就可以了
分为GGGHGG和GGHGGG这两部分
相关代码(该方法也是看了被人思路学会的)
#include<bits/stdc++.h>
using namespace std;
int main()
{
string s;
int n;
cin>>n;
cin>>s;
long long num;
if(int i=0;i<n;i++)
{
int zl=i-1,yl=i+1;//设zl为目前判定的左面的符号,yl是右边的符号
long long zn=0,yn=0;//zn是左边连续相同的个数,yn是右边连续相同的个数
while(s[i]!=s[zl]&&i>=1)zn++,zl--;//当左边和是s[i]不相等时, 一直往左判断,直到不相等。判断出左边连续相等字母的个数
while(s[i]!=s[yl]&&i<n-1)yn++,yl--;//同理, 判断出右边连续相等字母的个数
num+=zn*yn+max((long long)0,zn-1)+max((long long)0,yn-1);
}
cout<<num<<endl;
return 0;
}
现在我来一点点的分析:
比如输入的是:GGGHGGHGGG
前面有个for循环就是对它们一个个的检查
代码
while(s[i]!=s[zl]&&i>=1)zn++,zl--;//当左边和是s[i]不相等时, 一直往左判断,直到不相等。判断出左边连续相等字母的个数
通过这个while循环,s[i]!=s[zl]就是检索到左右不是同一个字母了,也就是例子中的H,当s[i]=='H'时,然后zn++,zl--,直到得出这个H的左边有几个G
代码
while(s[i]!=s[yl]&&i<n-1)yn++,yl--;//同理, 判断出右边连续相等字母的个数
这行代码,同理,得出这个H右边的G有几个
代码
num+=zn*yn+max((long long)0,zn-1)+max((long long)0,yn-1);
将三者相加
大致思路就是这样
写那么详细方便自己以后复习,本人目前菜鸟一个orz
不是我想出的这个方法