C++——红绿灯

红绿灯

Description

你发现自己在一个不寻常的十字路口，有一个奇怪的红绿灯。红绿灯有三种可能的颜色:红($r$)，黄($y$)，绿($g$)。众所周知，红绿灯每$n$秒重复一次颜色，在第$i$秒时，颜色$s_i$是亮的。
这样，颜色的顺序就由一个字符串来描述。例如，如果$\text{s="rggry"}$，那么交通灯的工作方式如下:红-绿-绿-红-黄-红-绿-绿-红-黄-… 等等。
更正式地说，你得到一个长度为$n$的字符串$s_1,s_2，\dots ，s_n$。在第一秒，颜色$s_1$是亮的，在第二秒，颜色$s_2$，…在第$n$秒时，颜色$s_n$亮起，在第$n+1$秒时，颜色$s_1$亮起，以此类推。
你需要过马路，只有绿灯亮着的时候才能过马路。
你知道现在的红绿灯是什么颜色，但你不知道现在的时间。你需要找出保证过马路的最短时间。
你可以假设你马上过了马路。
例如，$\text{s="rggry"}$和当前红绿灯颜色$r$有两个选项:绿色在$1$秒后或$3$秒后亮起。这样的话，答案就等于$3$，这是保证我们过马路的秒数，如果当前的颜色是$r$。

Format

Input

第一行包含整数$n$和符号$c$($c$为允许的红绿灯颜色$r$、$y$或$g$中的一种)——字符串$s$的长度和当前红绿灯的颜色。
第二行包含长度为$n$的字符串$s$，由字母$r$、$y$和$g$组成

Output

输出保证您过马路的最小秒数。

Samples

输入数据 1

5 r
rggry

输出数据 1

3

输入数据 2

9 y
rrrgyyygy

输出数据 2

4

Constraints

对于$50\%$的数据，$1\le n\le 10^3$
对于$100\%$的数据，$1\le n\le 2\ast 10^5$

---

题解

#include <bits/stdc++.h>
#define ll long long
using namespace std;
int t;
const int N=2e5+5;
int n;
char y;
string s;
int a[N*2],cnt;
void solve()
{
    cin>>n>>y>>s;
    s+=s;
    cnt=0;
    int mx=0;
    for(int i=0;i<s.size();i++){
        if(s[i]=='g') a[++cnt]=i;
    }
    for(int i=0;i<s.size()/2;i++){
        if(s[i]==y){
            int k=lower_bound(a+1,a+cnt+1,i)-a;
            mx=max(a[k]-i,mx);
        }
    }
    cout<<mx<<"\n";
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    t=1;
    solve();
    return 0;
}