CF 877B Nikita and string

B. Nikita and string

time limit per test2 seconds

memory limit per test256 megabytes

inputstandard input

outputstandard 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

inputCopy

abba

outputCopy

4

inputCopy

bab

outputCopy

2

Note

It the first sample the string is already beautiful.

In the second sample he needs to delete one of “b” to make it beautiful.

题解：前缀和优化

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e6 + 5;
char str[maxn];
int sum1[maxn], sum2[maxn];
/*
*/
int main() {
  scanf("%s", str + 1);
  int n = strlen(str + 1);
  for (int i = 1; i <= n; i++) {
    sum1[i] = sum1[i - 1];
    sum2[i] = sum2[i - 1];
    if (str[i] == 'a') sum1[i]++;
    if (str[i] == 'b') sum2[i]++;
  }
  int ans = 0;
  int mm = n;
  for (int i = 0; i <= n; i++) {
    mm = min(mm, sum2[i] - sum1[i]);
    ans = max(ans, sum2[i] - sum1[i] - mm);
  }
  cout << ans + sum1[n]<< endl;
  return 0;
}