bzoj1150: [CTSC2007]資料備份Backup(堆+貪心)

Hanks_o發表於2018-03-25

題目傳送門

解法:
跟上一題差不多一個思想。
bzoj2151
只不過這題多了一個比較噁心的邊界而已。

那麼相對於邊界值。就拿他前面沒有點來舉例。
假設現在前面沒有點了。那麼肯定選他之後不用反悔啊。
因為你後面那個肯定比你小啊要不然幹嘛選你呢。
所以說選完之後把自己和後面的點刪除就行了。
維護一下即可。
跟2151唯一的不同也就這個點了。
如果後面沒有點也是一樣的道理。。

程式碼實現:

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<queue>
#include<iostream>
using namespace std;
struct node {
    int w,c;node() {w=c=0;}
    friend bool operator <(node n1,node n2){return n1.c>n2.c;}
};priority_queue<node> a;
int q[110000],h[110000],n,m,s[110000];bool v[110000];
int main() {
    scanf("%d%d",&n,&m);
    scanf("%d",&s[1]);
    for(int i=2;i<=n;i++) {
        scanf("%d",&s[i]);q[i]=i-1;h[i]=i+1;
        node ss;ss.c=s[i]-s[i-1];ss.w=i;a.push(ss);
    }h[n]=-1;q[2]=-1;
    for(int i=n;i>=2;i--)s[i]-=s[i-1];s[1]=0;
    memset(v,false,sizeof(v));int ans=0;
    for(int i=1;i<=m;i++) {
        while(v[a.top().w]==true)a.pop();
        node t=a.top();a.pop();ans+=t.c;int l=q[t.w],r=h[t.w];
        node p;p.w=t.w;
        if(l!=-1&&r!=-1) {
            v[l]=true;p.c+=s[l];v[r]=true;p.c+=s[r];p.c-=t.c;
            s[t.w]=p.c;q[t.w]=q[l],h[q[l]]=t.w;h[t.w]=h[r],q[h[r]]=t.w;a.push(p);
        }
        else if(l==-1) {v[t.w]=true;v[h[t.w]]=true;q[h[h[t.w]]]=-1;}
        else {v[t.w]=true;v[q[t.w]]=true;h[q[q[t.w]]]=-1;}
    }printf("%d\n",ans);
    return 0; 
}

相關文章