题意:一个红绿灯以n秒为周期变换颜色,给定一个长度为n的字符串表示红绿灯单个周期的变换规律,再给定某时刻红绿灯的颜色,为字符a。判定最坏情况下需要经过几秒红绿灯才能变为绿色。
思路:首先特判,给定字符为g 则输出0
若颜色不为g,则从字符串首位枚举,判断若i字符等于a,则继续向右枚举找到g,记录两个字符的位置差值,大于原答案,则更新答案。
但, 可能字符a之后的下一个字符g在红绿灯的下一个周期出现,所以将题目给定字符串复制,并与原来的字符串拼接,实现2T个周期。
又,n<=2*10e5,tl=1sec,复杂度O(2n)需要优化。
优化1:我们寻找的是最坏情况,在i ----> g的枚举过程中,若中间再一次出现字符j==a,字符j与g差值必小于i与g,故无需判断.
优化2:若遍历过程中i的下标大于n且找到字符g,则无需继续遍历,break。
#include <bits\stdc++.h>
typedef int int_32;
#define int long long
using namespace std;
int_32 main(void) {
int t;
cin >> t;
while (t--) {
//信息维护,制造两个变量存储信息,一个flag表示是否找到需要的字符,找到后flag成为存储位置信息的容器
//另一个变量ans,负责更新答案
int n;
char a;
string s;
cin >> n >> a >> s;
//a为绿色,无需判断
if (a == 'g') {
cout << 0<<endl;
continue;
}
s = s + s;
int flag_pos = -1;
int ans = -1;
for (int i = 0; i < 2 * n; i++) {
if (s[i] == a && flag_pos == -1)flag_pos = i;
if (s[i] == 'g' && flag_pos != -1) {
ans = max(ans, i - flag_pos);
flag_pos = -1;
if (i > n)break;
}
}
cout << ans << endl;
}
}