Codeforces Round #442 (Div. 2) B题【一道模拟题QAQ】

B. Nikita and string

One day Nikita found the string containing letters "a" and "b" only.

Nikita thinks that string is beautiful if it can be cut into 3 strings (possibly empty) without changing the order of the letters, where the 1-st and the 3-rd one contain only letters "a" and the 2-nd contains only letters "b".

Nikita wants to make the string beautiful by removing some (possibly none) of its characters, but without changing their order. What is the maximum length of the string he can get?

Input
The first line contains a non-empty string of length not greater than 5 000 containing only lowercase English letters "a" and "b".

Output
Print a single integer — the maximum possible size of beautiful string Nikita can get.

Input

abba

Output

4

题意：给出一个字符串，这个字符串长度最大是5000，并且规定只有ab组成，现在问你，能不能将这个字符串分成三部分，第一和第三部分只有a或者是空，第二部分由b或者空组成。问你能不能分成这样的形式，如果能那么最长的长度是多少。

思路：暴力+模拟【只有5中情况。分别是a,b,ab,ba,aba。那么我们只要统计出这五种情况的最大的长度，最后我们取其中最大的就好了】

AC代码：{调了挺久的QAQ 菜是原罪}

#include<bits/stdc++.h>

using namespace std;
#define N 1200000
set<char> sss;
struct str{
    int type;
    int num;
    int qian;// 前缀 
    int hou;// 后缀 
}st[N];
int main(){
    string str;
    cin>>str;
    int suma=0;
    int sumb=0;
    int cnt=0;
    int add=1;
    for(int i=0;i<=str.size();i++){
        if(str[i]=='a'&&i!=str.size()){
            suma++;
        }else if(str[i]=='b'&&i!=str.size()){
            sumb++;
        }
        if(str[i]!=str[i+1]){
            st[cnt].type=str[i]-'a';
            st[cnt++].num=add;
            add=1;
        }else{
            add++;
        }
        if(i!=str.size())
            sss.insert(str[i]);
    }
    if(sss.size()==1){  //  a,b情况 
        printf("%d
",str.size());
        return 0;
    }
    int sa=0;
    int sb=0;
    for(int i=0;i<cnt;i++){// 前缀 
        if(st[i].type==1){
            st[i].qian=sa;
        }else{
            st[i].qian=sb;
        }
        if(st[i].type==1){ // 类型：B 
            sb+=st[i].num;
        }else{
            sa+=st[i].num;
        }
    }
    sa=0;
    sb=0;
    for(int i=cnt-1;i>=0;i--){
        if(st[i].type==1){
            st[i].hou=sa;
        }else{
            st[i].hou=sb;
        }
        if(st[i].type==1){ // 类型：B 
            sb+=st[i].num;
        }else{
            sa+=st[i].num;
        }
    }
    /*
    for(int i=0;i<cnt;i++){
        printf("%d ",st[i].num);
    }
    printf("
");
    for(int i=0;i<cnt;i++){
        printf("%d %d 
",st[i].qian,st[i].hou);
    }
    */
    int ans=max(suma,sumb);
    
    for(int i=0;i<cnt;i++){
        if(st[i].type==1){
            ans=max(ans,st[i].num+st[i].hou+st[i].qian);
            int add=0;
            for(int j=i+2;j<cnt;j+=2){
                add+=st[j].num;
                ans=max(ans,st[i].qian+add+st[i].num+st[j].hou);
            }
        }
        
    }
    for(int i=0;i<cnt;i++){
        ans=max(ans,st[i].num+st[i].hou);
        int res=0;
        for(int j=i+2;j<cnt;j+=2){
            res+=st[j].num;
            ans=max(ans,st[i].num+res+st[j].hou);
        }
    }
        
    cout<<ans;
    return 0;
}

/*
bbabbbbbaaba
aabaaaaabbab
aaaabaabbbbbaaabaaaa
a,b,ab,ba,aba 


bb a bbbbb aa b a

2  1   5   2  1 1




*/