COJ 2110 Day7-例3

Day7-例3

难度级别：C； 运行时间限制：5000ms； 运行空间限制：256000KB； 代码长度限制：2000000B

试题描述

 

输入

输入的第一行包含整数n和k，其中n（2 ≤ n ≤100 000）表示办公楼的数目，k（1≤ k≤ n/2）表示可利用的网络电缆的数目。接下来的n行每行仅包含一个整数（0≤ s ≤1000 000 000）, 表示每个办公楼到大街起点处的距离。这些整数将按照从小到大的顺序依次出现。

输出

输出应由一个正整数组成，给出将2K个相异的办公楼连成k对所需的网络电缆的最小总长度。

输入示例

5 2
1
3
4
6
12

输出示例

4

题解：奇妙的题。用堆维护每一段，如果取了一个就要把（两边的-这个）放到堆里，表示可以改成选两边的，但是就是不能同时选。

用set实现：

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<stack>
#include<queue>
#include<cstring>
#include<set>
#define PAU putchar(' ')
#define ENT putchar('
')
using namespace std;
const int maxn=100000+10;
const long long inf=4e18;
struct data{long long v;int p;};
bool operator<(const data&a,const data&b){return a.v<b.v||(a.v==b.v&&a.p<b.p);}
inline long long read(){
    long long x=0,sig=1;char ch=getchar();
    for(;!isdigit(ch);ch=getchar())if(ch=='-')sig=0;
    for(;isdigit(ch);ch=getchar())x=10*x+ch-'0';
    return sig?x:-x;
}
inline void write(long long x){
    if(x==0){putchar('0');return;}if(x<0)putchar('-'),x=-x;
    int len=0;long long buf[20];while(x)buf[len++]=x%10,x/=10;
    for(int i=len-1;i>=0;i--)putchar(buf[i]+'0');return;
}
set<data>S;
long long A[maxn],d[maxn];
int L[maxn],R[maxn],n,k;
int main(){
    data x;set<data>::iterator t;
    n=read();k=read();
    for(int i=1;i<=n;i++)A[i]=read();
    for(int i=1;i<n;i++)d[i]=A[i+1]-A[i];d[0]=inf;d[n]=inf;
    for(int i=0;i<n;i++)R[i]=i+1,L[i+1]=i;
    for(int i=0;i<=n;i++)S.insert((data){d[i],i});
    long long ans=0;
    for(int i=1;i<=k;i++){
        t=S.begin();x=*t;ans+=x.v;
        int l=L[x.p],r=R[x.p];
        S.erase(t);S.erase((data){d[l],l});S.erase((data){d[r],r});
        R[l]=R[r];L[R[r]]=l;d[l]+=d[r]-x.v;
        S.insert((data){d[l],l});
    }write(ans);return 0;
}

 treap仍然调不对啊。。。为什么。。。我改了重载也过不了啊。。。。。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<stack>
#include<queue>
#include<cstring>
#include<ctime>
#include<set>
#define PAU putchar(' ')
#define ENT putchar('
')
#define CH for(int d=0;d<2;d++)if(ch[d])
using namespace std;
const int maxn=200000+10,maxnode=7000000+10;
const long long inf=4e18;
struct data{long long v;int p;};
bool operator>(const data&a,const data&b){return a.v>b.v||(a.v==b.v&&a.p>b.p);}
bool operator==(const data&a,const data&b){return (a.v==b.v)&&(a.p==b.p);}
struct node{
    node*ch[2];data v;int siz,r;
    void init(){siz=1;r=rand();ch[0]=ch[1]=NULL;return;}
    void update(){siz=1;CH{siz+=ch[d]->siz;}return;}
}treap[maxnode],*nodecnt=treap,*root;queue<node*>RM;
node*newnode(){node*k;if(RM.empty())k=nodecnt++;else k=RM.front(),RM.pop();k->init();return k;}
void del(node*&x){RM.push(x);x=NULL;return;}
void rotate(node*&x,int d){
    node*k=x->ch[d^1];x->ch[d^1]=k->ch[d];k->ch[d]=x;x->update();k->update();x=k;return;
}
void insert(node*&x,data v){
    if(!x)x=newnode(),x->v=v;
    else{int d=v>x->v;insert(x->ch[d],v);
        if(x->r<x->ch[d]->r)rotate(x,d^1);else x->update();
    }return;
}
void remove(node*&x,data v){
    if(x->v==v){
        if(x->ch[0]&&x->ch[1]){
            int d=x->ch[0]->r>x->ch[1]->r;rotate(x,d);remove(x->ch[d],v);
        }else{node*k=x;x=x->ch[0]?x->ch[0]:x->ch[1];del(k);}
    }else remove(x->ch[v>x->v],v);
    if(x)x->update();return;
}
void print(node*x){
    if(!x)return;print(x->ch[0]);printf("%lld ",x->v.v);print(x->ch[1]);return;
}
data getmi(node*x){
    if(x->ch[0])x=x->ch[0];return x->v;
}
inline long long read(){
    long long x=0,sig=1;char ch=getchar();
    for(;!isdigit(ch);ch=getchar())if(ch=='-')sig=0;
    for(;isdigit(ch);ch=getchar())x=10*x+ch-'0';
    return sig?x:-x;
}
inline void write(long long x){
    if(x==0){putchar('0');return;}if(x<0)putchar('-'),x=-x;
    int len=0;long long buf[20];while(x)buf[len++]=x%10,x/=10;
    for(int i=len-1;i>=0;i--)putchar(buf[i]+'0');return;
}
long long A[maxn],d[maxn];
int L[maxn],R[maxn],n,k;
int main(){
    srand(time(0));
    n=read();k=read();
    for(int i=1;i<=n;i++)A[i]=read();
    for(int i=1;i<n;i++)d[i]=A[i+1]-A[i];d[0]=inf;d[n]=inf;
    for(int i=0;i<n;i++)R[i]=i+1,L[i+1]=i;
    for(int i=0;i<=n;i++)insert(root,(data){d[i],i});
    long long ans=0;
    for(int i=1;i<=k;i++){
        data t=getmi(root);ans+=t.v;
        int l=L[t.p],r=R[t.p];
        remove(root,t);remove(root,(data){d[l],l});remove(root,(data){d[r],r});
        R[l]=R[r];L[R[r]]=l;d[l]+=d[r]-t.v;
        insert(root,(data){d[l],l});
    }write(ans);return 0;
}