【Henu ACM Round #12 B】 Alice, Bob, Two Teams

【链接】 我是链接,点我呀:)
【题意】

在这里输入题意

【题解】

写个前缀和 和 一个后缀和。 (即前i个字符A所代表的数字的和以及前i个字符B所代表的数字的和。。 然后枚举前i个字符翻转。 求B对应数字的最大值

枚举i..n这些字符翻转
求B对应数字的最大值

两个同时求最大值就是答案了。

【代码】

#include <bits/stdc++.h>
#define ll long long
using namespace std;

const int N = 5e5;
int a[N+10],n,b[N+10];
ll pre[N+10][2],after[N+10][2];
string s;


int main(){
	#ifdef LOCAL_DEFINE
	    freopen("rush_in.txt", "r", stdin);
	#endif
	ios::sync_with_stdio(0),cin.tie(0);
    cin >> n;
    for (int i = 1;i <= n;i++) cin >> a[i];
    cin >> s;
    for (int i = 1;i <= n;i++) b[i] = s[i-1]-'A';

    for (int i = 1;i <= n;i++){
        pre[i][0] = pre[i-1][0];
        pre[i][1] = pre[i-1][1];
        if (b[i]==0) pre[i][0]+=a[i];
        if (b[i]==1) pre[i][1]+=a[i];
    }
    for (int i = n;i >= 1;i--){
        after[i][0] = after[i+1][0];
        after[i][1] = after[i+1][1];
        if (b[i]==0) after[i][0]+=a[i];
        if (b[i]==1) after[i][1]+=a[i];
    }
    ll ans = 0;
    for (int i= 1;i <= n;i++)
        if (b[i]==1) ans+=a[i];


    for (int i = 1;i <=n;i++)
        ans=max(ans,after[i+1][1]+pre[i][0]);

    for (int i = n;i >= 1;i--)
        ans = max(ans,after[i][0]+pre[i-1][1]);

    cout << ans << endl;
	return 0;
}