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;}