题意大意
给定一个字符串
s
s
s,定义
A
B
(
s
)
\mathrm{AB}(s)
AB(s) 为
s
s
s 中串 ab
出现的次数,
B
A
(
s
)
\mathrm{BA}(s)
BA(s),为 ba
出现的次数。
每次可以修改一个字符,要求通过尽量少的操作 ,使得 A B ( s ) = B A ( s ) \mathrm{AB}(s)=\mathrm{BA}(s) AB(s)=BA(s)。
输出修改后的字符串。
T T T 组数据。
1 ≤ T ≤ 1000 1\le T\le 1000 1≤T≤1000, 1 ≤ ∣ s ∣ ≤ 100 1 \leq |s| \leq 100 1≤∣s∣≤100。
解题思路
结论题。
先计算 A B ( s ) \mathrm{AB}(s) AB(s) 和 B A ( s ) \mathrm{BA}(s) BA(s),分别记为 c c c 和 d d d。
然后,通过手推,可以发现 0 ≤ ∣ c − d ∣ ≤ 0 \leq |c-d| \leq 0≤∣c−d∣≤ 1。(显然)。
如果 ∣ c − d ∣ = 0 |c-d|=0 ∣c−d∣=0,直接输出原字符串即可。
否则,我们可以通过一次修改获得答案字符串。
先找到字符串中相邻相等的那些串,直接删去,留下一个,因为不影响 c c c 和 d d d 的取值。
例如:
aaabbbaba
和 ababa
的
c
c
c 和
d
d
d 是分别相等的。
又例如:
baaaba
和 baba
的
c
c
c 和
d
d
d 是分别相等的。
注意,上面所说的删除操作只是纸上模拟便于思考,在代码中不用写。
所以,执行上面操作后,肯定可以得到 ababab...
或 bababa...
这两种串中的一个的。
然后怎么修改就不用说了吧。
如果是 ``ababab…这种的,肯定是 $c=d+1$,将第一个字符改为
b` 即可。
如果是 ``bababa…这种的,肯定是 $d=c+1$,将第一个字符改为
a` 即可。
CODE
#include <bits/stdc++.h>
using namespace std;
int T;
string s;
signed main()
{
scanf("%d", &T);
while(T--)
{
int ab = 0, ba = 0;
cin >> s;
for(int i = 1; i < s.size(); ++i)
{
if(s[i] == 'a' && s[i - 1] == 'b') ba++;
if(s[i] == 'b' && s[i - 1] == 'a') ab++;
}
if(ab == ba)
{
cout << s << "\n";
continue;
}
if(s[0] == 'a') s[0] = 'b';
else s[0] = 'a';
cout << s << "\n";
}
return 0;
}