BZOJ1150(CTSC2007)[数据备份Backup]--贪心+链表+堆

【链接】
bzoj1150

【解题报告】

对于最优解肯定是相邻两点建边最优。所以我们只需要选最小的前$K$段。

对于上述的贪心，有如下情况：

从x===x==x===x选两段。

如果直接选最小的肯定会选中间这段(x===x**==**x===x)。

但是实际上只能选两边(x**===x==x===**x)。

经过考虑，我们可以发现如果不选从堆里挑出来前$K$段中的某一段，一定会选这段两边的两段。

所以我们只需要再用链表维护一下，就可以了。

#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
const int maxn=100005;
int n,m,ans,dst[maxn],L[maxn],R[maxn];
struct Data
{
    int s,x;
    Data (int p,int i) {s=p; x=i;}
    bool operator < (const Data &a) const{
        return s>a.s;
    }
};
priority_queue<Data> hep;
inline char nc()
{
    static char buf[100000],*l,*r;
    if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
    if (l==r) return EOF; return *l++;
}
inline int Read()
{
    int res=0,f=1; char ch=nc(),cc=ch;
    while (ch<'0'||ch>'9') cc=ch,ch=nc();
    if (cc=='-') f=-1;
    while (ch>='0'&&ch<='9') res=res*10+ch-48,ch=nc();
    return res*f;
}
int main()
{
    freopen("1150.in","r",stdin);
    freopen("1150.out","w",stdout);
    n=Read(); m=Read(); ans=0;
    while (!hep.empty()) hep.pop();
    for (int i=1,x,lst=0; i<=n; i++) x=Read(),dst[i]=x-lst,lst=x;
    for (int i=2; i<=n; i++) hep.push((Data){dst[i],i}),L[i]=i-1,R[i]=i+1; L[2]=R[n]=0;
    for (int i=1; i<=m; i++)
    {
        Data now=hep.top();
        while (now.s!=dst[now.x]) hep.pop(),now=hep.top();
        int l=L[now.x],r=R[now.x]; hep.pop(); ans+=now.s;
        if (!l) {if (R[r]) L[R[r]]=0; dst[r]=-1; continue;}
        if (!r) {if (L[l]) R[L[l]]=0; dst[l]=-1; continue;}
        dst[now.x]=dst[l]+dst[r]-dst[now.x]; hep.push((Data){dst[now.x],now.x});
        L[now.x]=L[l]; R[now.x]=R[r];
        if (L[l]) R[L[l]]=now.x;
        if (R[r]) L[R[r]]=now.x;
        dst[l]=dst[r]=-1;
    }
    printf("%d",ans);
    return 0;
}