Description
Alice and Bob are playing a game. The game involves splitting up game pieces into two teams. There are n pieces, and the i-th piece has a strength pi.
The way to split up game pieces is split into several steps:
- First, Alice will split the pieces into two different groups A and B. This can be seen as writing the assignment of teams of a piece in an n character string, where each character is A or B.
- Bob will then choose an arbitrary prefix or suffix of the string, and flip each character in that suffix (i.e. change A to B and B to A). He can do this step at most once.
- Alice will get all the pieces marked A and Bob will get all the pieces marked B.
The strength of a player is then the sum of strengths of the pieces in the group.
Given Alice's initial split into two teams, help Bob determine an optimal strategy. Return the maximum strength he can achieve.
Input
The first line contains integer n (1 ≤ n ≤ 5·105) — the number of game pieces.
The second line contains n integers pi (1 ≤ pi ≤ 109) — the strength of the i-th piece.
The third line contains n characters A or B — the assignment of teams after the first step (after Alice's step).
Output
Print the only integer a — the maximum strength Bob can achieve.
Sample Input
5 1 2 3 4 5 ABABA
11
5 1 2 3 4 5 AAAAA
15
1 1 B
1
Hint
In the first sample Bob should flip the suffix of length one.
In the second sample Bob should flip the prefix or the suffix (here it is the same) of length 5.
In the third sample Bob should do nothing.
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
char str[500010];
int p[500010];
int main()
{
int n;
while(scanf("%d",&n)!=EOF)
{
for(int i=0;i<n;i++)
{
scanf("%d",&p[i]);
}
scanf("%s",str);
long long ans=0,sum=0;
for(int i=0;i<n;i++)
{
if(str[i]=='B')
{
ans+=p[i];//先记录没翻转时的B的价值和;
}
}
long long temp=ans;
sum=ans;//sum记录最后结果;
for(int i=0;i<n;i++)//先向前翻转;
{
if(str[i]=='B')//遇到B减去,遇到A加上;
{
temp-=p[i];
}
else if(str[i]=='A')
{
temp+=p[i];
}
sum=max(temp,sum);
}
long long cnt=ans;
for(int i=n-1;i>=0;i--)//再向后翻转;
{
if(str[i]=='B')
{
cnt-=p[i];
}
else if(str[i]=='A')
{
cnt+=p[i];
}
sum=max(cnt,sum);
}
printf("%lld\n",sum);
}
return 0;
}