洛谷P10115 [LMXOI Round 1] Placer

2301_82345937

已于 2024-02-03 17:40:19 修改

阅读量707

点赞数 15

文章标签：动态规划图论算法

于 2024-02-03 17:32:07 首次发布

本文链接：https://blog.csdn.net/2301_82345937/article/details/136007328

版权

文章讲述了如何使用动态规划解决给定括号序列的问题，通过优化转移方程，找到最大美丽度。关键在于利用定值a_i简化计算，引入c_i并利用栈来存储匹配关系，从而降低时间复杂度。

摘要由CSDN通过智能技术生成

题意：给定一个长度为 $n$ 的括号序列 $S$ ,一个长度为 $n$ 的序列 $a_i$ 。定义一段括号序列的价值：若括号序列 $[l, r]$ 为合法序列，其价值为 $a_r-a_l$ ;若不合法，价值为 $0$ .可将 $S$ 划分为若干非空子段，定义美丽度为每个子段价值之和，求最大美丽度。 $n\le3000000$ .

不难想到dp，设 $f_i$ 表示原括号序列的前i位所能得到的最大美丽度。转移方程为 $f_i = max(f_{i-1}, f_j+a_i-a_{j+1})$ 其中 $j\in[1, i-2]$ ,并且 $[j + 1, i]$ 是一个合法的括号序列。
这样可以得到 $O(n^2）$ 的算法。
代码：

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

typedef long long ll;
#define For(i, a, b) for (int i = (a); i <= (b); i++)
#define FOR(i, a, b) for (int i = (a); i >= (b); i--)

const int N = 3e6 + 5;

int n, a[N];
ll dp[N];
char s[N];

int main() {
    scanf("%d", &n);
    scanf("%s", s + 1);
    For (i, 1, n) scanf("%d", &a[i]);
    memset(dp, 0xcf, sizeof(dp));
    dp[0] = 0;
    For (i, 1, n) {
        dp[i] = dp[i - 1];
        if (s[i] == '(') continue;
        else {
            int res = 1;
            FOR(j, i - 1, 1) {
                if (s[j] == '(') res -= 1;
                else res += 1;
                if (res < 0) break;
                if (!res) dp[i] = max(dp[i], dp[j - 1] + (ll)a[i] - a[j]);
            }
        }
    }
    printf("%lld\n", dp[n]);
    return 0;
}

考虑优化。
注意 $a_i$ 为定值，所以找 $max\{ f_j+a_i-a_{j+1}\}$ 就是找
$max\{f_j-a_{j+1}\}$
记 $c_i=max\{f_j-a_{j+1}\}$ , $[j + 1, i]$ 为合法括号序列。
那么 $f_i=max\{c_i+a_i,f_{i-1}\}$ ,只要快速计算 $c_i$ 即可。
求 $c_i$ 的方法比较精彩，对于一个位置在 $i$ 的右括号，用 $lst_i$ 记录能与其匹配的最近左括号位置。这步操作可以用一个类似栈的方法做到。则 $c_i=max\{c_{lst_i-1, f_{lst_i-1} -a_{lst_i}}\}$
代码：

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

typedef long long ll;
typedef pair<int, int> pii;
#define fi first
#define se second
#define pb push_back
#define mp make_pair
#define For(i, a, b) for (int i = (a); i <= (b); i++)
#define FOR(i, a, b) for (int i = (a); i >= (b); i--)

const int N = 3e6 + 5;

int n, a[N], lst[N], stk[N];
ll f[N], c[N];
char s[N];

int main() {
    scanf("%d", &n);
    scanf("%s", s + 1);
    For (i, 1, n) scanf("%d", &a[i]);
    memset(f, 0xcf, sizeof(f));
    f[0] = 0;
    memset(c, 0xcf, sizeof(c));
    int top = 0;
    For (i, 1, n) {
        if (s[i] == ')' && top) lst[i] = stk[top--];
        else if (s[i] == '(') stk[++top] = i;
    }
    For (i, 1, n) {
        f[i] = f[i - 1];
        if (s[i] == ')' && lst[i]) {
            c[i] = max(c[lst[i] - 1], f[lst[i] - 1] - a[lst[i]]);
            f[i] = max(c[i] + a[i], f[i]);
        }
    }
    printf("%lld\n", f[n]);
    return 0;
}