AB Balance

Codeforce 1606.AB Balance

AB Balance

题解

翻译一下意思,就是在一堆全是a和b的字符串里要使ab和ba的数量相同,否则修改a变成b或者b变成a。
由于当中间每出现一个aba或者bab时,ab和ba的数量都会增加一,所以只用考虑字符串前后的字符。

#include<bits/stdc++.h>
using namespace std;
int n;
int main(){
    ios::sync_with_stdio(0);
    cin.tie(NULL);cout.tie(NULL);
    cin>>n;
    while(n--){
        string s;
        cin>>s;
        int ab=0,ba=0;
        for(int i=0;i<s.size();i++) {
            if(s[i]=='a'&&s[i+1]=='b') ab++;
            if(s[i]=='b'&&s[i+1]=='a') ba++;
        }
        if(ab>ba){
            int pos=s.find('a');
            s[pos]='b';
        }
        else if(ab<ba){
            int pos=s.find('b');
            s[pos]='a';
        }
        cout<<s<<endl;
    }
    return 0;
}

本题结束,完结撒花撒比竟是我自己。

后来看到了wock的代码思路……震惊,原来改动s[0]就可以做到满足条件!
众说周知,cf的a题,输出结果是可以有很多种,然后输出任意一个就ok,往往我们可以去找一个比较特殊的解答直接输出。
be like

if(ab==ba) cout<<s<<endl;
else{
  if(a[0]=='a') s[0]='b';
  else s[0]='a';
  cout<<s<<endl;
}

题外话

每次刷题都能记录一个脑残小错误。
比如对s进行搜索时,i=0开始,到<s.size()结束。由于手比脑子快,又写成了i=1;i<=n,导致ab和ba的统计值一直不对。
差不多得了.jpg

上一篇:codeforce A. AB Balance


下一篇:位运算