CodeForces - 877B Nikita and string 【枚举区间】

B. Nikita and string

time limit per test

2 seconds

memory limit per test

256 megabytes

input

standard input

output

standard output

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.

Examples

input

abba

output

4

input

bab

output

2

题意：

一个字符串，如果可以分为三部分，第一、三部分只含a，第二部分只含b，可以为空，则称这个字符串为美丽字符串。输入一个只含a、b的字符串，判断删去一些字符后，可以组成的美丽字符串的最大长度。

思路：

可以用三个数组分别记录字符串中每一个元素前面a的个数pre[]，后面a的个数suf[]，前面b的个数b[]。这些都是一层循环可以解决的。然后就开始枚举第二部分的区间，需要两层循环。那么美丽字符串的长度=左边界的pre[]值+区间内的b的个数+右边界的suf[]值。其中区间内b的个数可以用b[R]-b[L-1]计算。

代码：

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<map>
#include<queue>
#include<ctype.h>
using namespace std;
char str[5005];
int pre[5005],suf[5005],b[5005];
int main()
{
    while(~scanf("%s",str))
    {
        memset(pre,0,sizeof pre);
        memset(suf,0,sizeof suf);
        memset(b,0,sizeof b);
        for(int i=0;i<strlen(str)+1;i++)
        {
            pre[i]=pre[i-1];
            if(str[i]=='a') pre[i]++;
            b[i]=b[i-1];
            if(str[i]=='b') b[i]++;
        }
        for(int i=strlen(str)-1;i>=0;i--)
        {
            suf[i]=suf[i+1];
            if(str[i]=='a') suf[i]++;
        }
        int maxlen=0;
        for(int l=0;l<strlen(str)+1;l++)
        {
            for(int r=l;r<strlen(str)+1;r++)
            {
                if(l==0) maxlen=max(maxlen,b[r]+suf[r+1]);
                else maxlen=max(maxlen,pre[l-1]+b[r]-b[l-1]+suf[r+1]);
            }
        }
        printf("%d\n",maxlen);
    }
    return 0;
}