51nod1053&&1052 最大M子段和

题面

我们先将所有连续正段和连续负段合并,那么选取所有正段一定是最优的,但是选取的段数有可能超过$m$段,这时我们就需要合并。

1.选取不在两端的一个负段与它两边正段合并,块数减少$1$,子段和减少$|v[i]|$。

2.选取一个正段,将其删除,并与其左右负段合并,块数减少$1$,子段和减少$|v[i]|$。

所以我们将每段按照$|v[i]|$大小排序,每次贪心合并至段数小于等于$m$。

#include <iostream>
#include <stdio.h>
#include <queue>
#include <algorithm>
using namespace std;
#define pa pair<int,int>
#define maxn 100001
#define INF 0x3f3f3f3f
inline int read()
{
    int s=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9')
    {
        if(ch=='-')
            f=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
        s=(s<<1)+(s<<3)+ch-'0',ch=getchar();
    return s*f;
}
int pr[maxn],nx[maxn],s[maxn];
int n,k,ans;
priority_queue<pa,vector<pa>,greater<pa> >q;
int main()
{
    n=read();
    k=read();
    int x,y;
    for(int i=1;i<=n;i++)
    {
        x=read();
        s[i]=x-y;
        y=x;
        if(i!=1)
            q.push(pa(s[i],i));
        pr[i]=i-1;
        nx[i]=i+1;
    }
    pr[2]=0;
    nx[n]=0;
    while(k--)
    {
        while(q.top().first!=s[q.top().second])
            q.pop();
        int u=q.top().second;
        int l=pr[u],r=nx[u];
        q.pop();
        ans+=s[u];
        s[u]=l&&r?s[l]+s[r]-s[u]:INF;
        pr[nx[u]=nx[r]]=u;
        nx[pr[u]=pr[l]]=u;
        s[l]=s[r]=INF;
        q.push(pa(s[u],u));
    }
    printf("%d",ans);
}