赛场上队友对着这题狂码130余行,贡献全场1/4的段错误,还是太年轻了。
思路是DP
将第i个左括号放在j及j右边能获得的最大收益记为状态。
i号‘(’ 放在 j位 ,则i右边的左括号都在j+1及以后的位置
或者i不放到j位,只沿用i放到j+1时的结果。
状态转移方程为:
dp[i][j] = max(dp[i+1][j+1]+a[I]*(i右边 i到j的位置差个右括号权值和),dp[i][j+1])
边界处理:全为0
答案:全dp数组中的最大值
循环框架:i(n~1)j(i能达到的最右位置~i的位置1)(i能达到的最右位置~i的位置)
#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn = 1010;
typedef long long LL;
int t, n;
char s[maxn];
LL a[maxn];
LL cnt[maxn];//位置i右边有几个右括号
//LL poi[maxn];//右边有i个右括号的位置为poi[i];包括
LL sum[maxn];//sum[poi[i]] 最右i个括号的权值和
LL dp[2][maxn];
int main() {
cin >> t;
while (t--) {
memset(dp, 0, sizeof(dp));
memset(sum, 0, sizeof(sum));
memset(cnt, 0, sizeof(cnt));
scanf("%d%s", &n, s+1);
for (int i = 1; i <= n; i++) scanf("%lld", &a[i]);
for (int i = n; i >= 1; i--) {
cnt[i] = cnt[i + 1] + (s[i] == ')' ? 1 : 0);
}
for (int i = n; i >= 1; i--) {
if (cnt[i] != cnt[i + 1]) {
sum[cnt[i]] = sum[cnt[i] - 1] + a[i];
}
}
int now = 0;
LL ans = 0;
for (int i = n; i >= 1; i--) {
if (s[i] == '(') {
for (int j = i + cnt[i]; j >= i; j--) {
dp[now ^ 1][j] = dp[now][j + 1] + a[i] * (sum[cnt[i]] - sum[cnt[i]-(j-i)]);
ans = max(ans, dp[now ^ 1][j]);
}
for (int j = i + cnt[i] - 1; j >= 1; j--) { //要记得-1
dp[now ^ 1][j] = max(dp[now ^ 1][j], dp[now ^ 1][j + 1]);
ans = max(ans, dp[now ^ 1][j]);
}
now ^= 1;
}
}
printf("%lld\n", ans);
}
//system("pause");
}