POJ 3017 DP + 单调队列

题意

传送门 POJ 3017

题解

设 $a_i$ 的前缀和为 $su{m}_i$ ，则有递推式

d p [ i ] = m i n s u m [ i ] − s u m [ j ] ≤ M { d p [ j ] + m a x j + 1 ≤ k ≤ i { a k } } dp[i]=min_{sum[i]-sum[j]\leq M}\{dp[j]+max_{j+1\leq k\leq i}\{a_k\}\} dp[i]=minsum[i]−sum[j]≤M​{dp[j]+maxj+1≤k≤i​{ak​}}

暴力求解 $O(N^2)$，需要优化。易推，$dp[i]$ 单调不减；设满足 $sum[i]-sum[j]<M$ 的区间 $[j+1,i]$ 元素最大值下标为 $idx$，则对于区间 $[j+1,idx)$，最小的 ( d p [ j ] + m a x j + 1 ≤ k < i d x { a k } ) (dp[j]+max_{j+1\leq k< idx}\{a_k\}) (dp[j]+maxj+1≤k<idx​{ak​}) 为 $(dp[j+1]+a_k)$；对于 $[idx,i]$ 区间的处理同理。

设满足 $sum[i]-sum[lb]\le M$ 的最小下标为 $lb$，对于元素 $a_i$，用满足 $sum[head]-sum[tail-1]\le M$ 的单调递减队列维护 $[lb,i]$ 内各个子区间的最大值；遍历单调队列更新 $dp[i]$ 即可。

#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <limits>
using namespace std;
#define min(a, b) ((a) < (b) ? (a) : (b))
#define max(a, b) ((a) > (b) ? (a) : (b))

typedef long long ll;
#define maxn 100005
ll n, m, sum[maxn], dp[maxn];
int a[maxn], idx[maxn];

int main()
{
    scanf("%lld%lld", &n, &m);
    bool f = 0;
    for (int i = 1; i <= n; i++)
    {
        scanf("%d", a + i);
        f |= (a[i] > m);
        sum[i] = sum[i - 1] + a[i];
    }
    if (f)
    {
        printf("%d\n", -1);
        return 0;
    }
    int l = 0, r = -1, lb = 1;
    for (int i = 1; i <= n; i++)
    {
        while (l <= r && sum[i] - sum[idx[l] - 1] > m) ++l;
        while (sum[i] - sum[lb-1] > m) ++lb;
        while (l <= r && a[idx[r]] <= a[i]) --r;
        idx[++r] = i, dp[i] = dp[lb - 1] + a[idx[l]];
        for (int j = l + 1; j <= r; j++)
        {
            dp[i] = min(dp[i], dp[idx[j - 1]] + a[idx[j]]);
        }
    }
    printf("%lld\n", dp[n]);
    return 0;
}