POJ 3017 Cut the Sequence(dp+单调队列)

202 篇文章 1 订阅

Description
给出一个长度为n的序列,要求将这个序列划分成若干块,使得每块的和不超过m,且每块的最大值之和最小
Input
第一行为两个整数n和m分别表示序列长度和划分块的和上限,之后n个整数表示这个序列
Output
输出各块最大值之和的最小值,如果不存在满足条件的划分方案则输出-1
Sample Input
8 17
2 2 2 8 1 8 2 1
Sample Output
12
Solution
用dp[i]表示序列前i个元素分完块后的最大值之和的最小值,那么显然可以得到一个递推方程dp[i]=min{dp[j]+max{a[k]}}(t<=j< i,j< k<=i)
但这个递推方程是O(n^2),所以需要优化,维护一个单调递增的单调队列,保证队列中所有元素之和不超过m,那么每进队一个元素,就可以用队列中元素的dp值推出这个元素的dp值
Code

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
#define maxn 1111111
#define INF 0x7ffffffff
typedef long long ll;
int n,que[maxn],p1,p2,k,flag;
ll m,a[maxn],cnt,dp[maxn];
int main()
{
    scanf("%d%lld",&n,&m);
    cnt=0,p1=1,p2=0,dp[0]=0,k=0,flag=1;
    for(int i=1;i<=n;i++)
    {
        scanf("%lld",&a[i]);
        if(a[i]>m)
        {
            flag=0;
            break;
        }
        cnt+=a[i];
        while(cnt>m)cnt-=a[++k];//保证队列中元素之和不超过m,k之前(包括k)元素都需出队 
        while(p2>=p1&&a[que[p2]]<=a[i])p2--;//维护单调性 
        que[++p2]=i;//新元素进队 
        while(p2>=p1&&que[p1]<=k)p1++;//k之前(包括k)的元素出队 
        dp[i]=INF;
        int tk=k;
        for(int j=p1;j<=p2;j++)
        {
            ll temp=dp[tk]+a[que[j]];
            if(dp[i]>temp)dp[i]=temp;
            tk=que[j];
        }
    }
    if(flag)printf("%lld\n",dp[n]);
    else printf("-1\n");
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值