AtCoder Beginner Contest 234 G - Divide a Sequence

AtCoder Beginner Contest 234 G - Divide a Sequence

题目

给定一个长度为N的序列，任意的将其划分为任意长度大于零的若干个子串，对于任意一个子串$S_i$有一个权值$max(S_i)-min(S_i)$，将其设为$V_i$,对于整个序列，其权值$W=V_1\ast V_2\ast V_3\dots \ast V_k$,现求对于长度为N的序列，所有可能划分产生的$W$的和$Sum$

思路

我们用DP的思想思考这个问题，如果我们知道了长$1$到$i-1$的序列的$Sum$，那么对于长度为$i$的序列，其和前状态本质在于所有加入$a_i$，的子串，对于这样一个子串，我们设其为{$a_j,a_{j+1},a_{j+2},\dots ,a_i$},我们可以算出它的$v$，可得
$$dp[i]=dp[i]+dp[i-j-1]\ast v$$
枚举j，可以得到朴素的转移方程
$$d{p}_i=\sum _{j=0}^{i}d{p}_j\ast \max (a_{j+1},a_{j+2},\dots ,a_i)-d{p}_j\ast \min (a_{j+1},a_{j+2},\dots ,a_i)$$
如果直接暴力枚举的时间复杂度是$O(n^2)$的，无法通过本题，考虑优化

首先，$max$和$min$可以分开计算
$$d{p}_i=\sum _{j=0}^{i}d{p}_j\ast \max (a_{j+1},a_{j+2},\dots ,a_i)-\sum _{j=0}^{i}d{p}_j\ast \min (a_{j+1},a_{j+2},\dots ,a_i)$$
对于一段后缀最值相关可以尝试用单调栈优化
观察以下dp转移方程

    dp[1] = dp[0] * (max(a[1]) - min(a[1]))
    
    dp[2] = dp[0] * (max(a[1],a[2]) - min(a[1],a[2])) 
          + dp[1] * (max(a[2]) - max(a[1]))
 
    dp[3] = dp[0] * (max(a[1],a[2],a[3]) - min(a[1],a[2],a[3]))
          + dp[1] * (max(a[2],a[3]) - min(a[2],a[3]))
          + dp[2] * (max(a[3]) - min(a[3]))

可以发现，答案都是一段后缀最值乘以某个dp值，最后对这些值求前缀和的过程，但每一次加入新的元素，这个前缀和的某些项就会发生改变。
这个过程可以说是用单调栈维护区间最值的同时，动态维护一段前缀和$mxsum$的过程。（对于最小值同理）
$$mxsum=\sum _{j=0}^{i}d{p}_j\ast \max (a_{j+1},a_{j+2},\dots ,a_i)$$
考虑如何用单调栈维护$mxsum$，我们设$mxsum$的任意项为$s_i$
对于单调栈，它维护的是第i个元素作为后缀最值的最长后缀$[l,i]$，那么$[l,i]$的所有后缀子段都会把最值修改为$a_i$，也即我们要更新的$s_i$。（或者说是由$a_i$，所统领的所有$s$)

因此每次弹出栈内元素的过程中，如果我们知道弹出元素所统领的$s_i$中的$dp[]$值都是哪些，就可以完成对$mxsum$的一次更新。

统计这些$dp[]$值，同样可以通过定义一个这样的$dp$数组:$dpm{x}_i$表示由第i个元素所统领的$s$所属的$dp[]$的和。
可写出转移方程：
$$dpm{x}_i=dpm{x}_i+dpm{x}_j$$
而j就是单调栈弹出的元素的下标。
这样我们就可以在单调栈维护过程中顺势维护了$mxsum$。对于$mnsum$同理。

最后，成功的把时间复杂度优化到了$O(n)$。

AC代码

#include <bits/stdc++.h>
#define yes puts("yes");
#define inf 0x3f3f3f3f
#define linf 0x3f3f3f3f3f3f3f3f
#define ll long long
#define ull unsigned long long
#define debug(x) cout<<"> "<< x<<endl;
#define endl '\n'
#define lowbit(x) x&-x
//#define int long long
using namespace std;
typedef pair<int,int> PII;
const int MAXN =10 + 2e5, mod = 998244353;

void solve()
{    
    int N;
    cin >> N;
    vector<int> a(N + 1);
    for(int i = 1;i <= N;i ++) cin >> a[i];
    // dp[i] = sum(dp[j] * max(a[j + 1 ~ i])) - sum(dp[j] * min(a[j + 1 ~ i]))
    vector<ll> dp(N + 1,0),dpmx(N + 1,0),dpmn(N + 1,0);
    dp[0] = dpmx[0] = dpmn[0] = 1;

    stack<int> Mx,Mn;
    int mxsum = 0,mnsum = 0;
    for(int i = 1;i <= N;i ++){
        dpmx[i] = dp[i - 1];// 长为1的后缀一定由ai统领
        while(Mx.size() && a[Mx.top()] < a[i]) {
            int id = Mx.top();
            Mx.pop();
            dpmx[i] = (dpmx[i] + dpmx[id]) % mod;// 求须修改项的dp[]和
            mxsum = (mxsum - dpmx[id] * a[id] % mod + mod) % mod;// 先减掉须修改值
        }
        Mx.push(i);
        mxsum = (mxsum + dpmx[i] * a[i] % mod + mod) % mod;// 加上修改后的值
        // 完成一次对mxsum的维护 
        dpmn[i] = dp[i - 1];
        while(Mn.size() && a[Mn.top()] > a[i]) {
            int id = Mn.top();
            Mn.pop();
            dpmn[i] = (dpmn[i] + dpmn[id]) % mod;
            mnsum = (mnsum - dpmn[id] * a[id] % mod + mod) % mod;
        }
        Mn.push(i);
        mnsum = (mnsum + dpmn[i] * a[i] % mod + mod) % mod;

        dp[i] = (mxsum - mnsum + mod) % mod;
    }
    cout << dp[N] << endl;
}
signed main()
{
    ios::sync_with_stdio();cin.tie();cout.tie();

    solve();

    return 0;
}

后记

算是我的单调栈和单调队列入门题了···，从知道这个概念到AC此题对我而言真的是个很困难的过程。中间查了资料，写了模板题，才回来理清这题的思路。十分曲折了可以说。所以尽力写好这篇博客。可最后还是有些不尽人意，感觉还有很多地方说的啰嗦或者不够切中要害，还是对单调栈和DP的理解不够深入。