BZOJ 1150 数据备份Backup

算是套路的贪心,因为给出数据是有序的,先将相邻的搞成线段存进堆里,每次取出一个最小的值,加到总值中,并改变左右的链表值,并将左右线段值存入当前位置作为撤销操作。

#include 
#include 
#include 
#define mk(a,b) make_pair(a,b) 
using namespace std;
const int inf=0x3f3f3f3f;
inline void read(int &tmk)  
{
    tmk=0;char ch=getchar();  int fu=1;  
    for (;ch<'0'||ch>'9';ch=getchar()) if (ch=='-') fu=-1;  
    for (;ch>='0'&&ch<='9';ch=getchar()) tmk=tmk*10+ch-'0';tmk*=fu;  
}
const int N=100005;
typedef pair<int,int> pii;
priority_queuevector,greater > q;  
int n,m;
int e[N],pre[N],nxt[N];
int main()
{
    register int i,j;
    read(n),read(m);
    int x=0,y;
    for (i=1;i<=n;i++)
    {
        read(y);
        e[i]=y-x;pre[i]=i-1,nxt[i]=i+1;x=y;
    }
    int ans=0;
    for (i=2;i<=n;i++) q.push(mk(e[i],i));
    pre[2]=0,nxt[n]=0;
    for (i=1;i<=m;i++)
    {
        while (q.top().first!=e[q.top().second])  q.pop();  
        int x=q.top().second,l=pre[x],r=nxt[x];  
        ans+=e[x];  q.pop();  
        pre[nxt[x]=nxt[r]]=x;  
        nxt[pre[x]=pre[l]]=x;  
        e[x]=l&&r?min(inf,e[l]+e[r]-e[x]):inf;  
        e[l]=e[r]=inf;
        q.push(mk(e[x],x));
    }
    printf("%d\n",ans);
    return 0;
}  

你可能感兴趣的:(bzoj,贪心,堆)