【链表】bzoj 1150: [CTSC2007]数据备份Backup

1150: [CTSC2007]数据备份Backup

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 1136  Solved: 458
[Submit][Status][Discuss]

Description

Input

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

Output

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

Sample Input

5 2
1
3
4
6
12

Sample Output

4

HINT

上面的样例输入给出了前面描述的示例情形 对于每一个测试点，如果写到输出文件中的答案正确，则得到该测试点100%的分数，否则得零分。30%的输入数据满足n≤20。60%的输入数据满足n≤10 000。

---

　　一开始觉得要贪心什么的

　　瞄了一眼题解

　　貌似我再想一下能出来( 大雾

　　然后感觉好困。。

　　最后几乎是半抄完了

　　易得每次只能选相邻点

　　所有只有n-1条边可选

　　每次链表选最小的

　　然后选完后合并前一个和后一个

　　具体来说就是堆维护最小值

　　每次合并-->该边和上一条和下一条不可选，创建一个新的节点代替原来的三个，ans加上权值

　　权值是左边+右边-中间(再选新节点=不选中间选两边)

　　还有就是选队首队尾，则易得选队首的后一个和选队尾的前一个不可能优。

 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #include<cmath>
 #include<queue>

 #define maxn 1000001

 using namespace std;

 int a[maxn*],n,k,pre[maxn*],next[maxn*],num=;

 long long cha[maxn*],ans=;

 bool vis[maxn*];

 struct node{
     int x;
     long long c;
     bool operator <(const node &A)const{
         if(A.c==c)return x>A.x;
         return c>A.c;
     }
 };

 priority_queue<node>q;

 inline int in()
 {
     int x=;char ch=getchar();
     while(ch<''||ch>'')ch=getchar();
     while(ch>=''&&ch<='')x=x*+ch-'',ch=getchar();
     return x;
 }

 void Pre()
 {
     for(int i=;i<n;i++)
         cha[i]=a[i+]-a[i],q.push((node){i,a[i+]-a[i]}),pre[i]=i-,next[i]=i+;
     num=n;
     pre[n]=n-;next[]=;
 }

 void solve()
 {
     node Top;
     for(int i=;i<=k;i++)
     {
         Top=q.top();q.pop();
         while(vis[Top.x])
             Top=q.top(),q.pop();
         if(pre[Top.x]==)
         {
             pre[next[next[Top.x]]]=;
             ans+=Top.c;
             vis[Top.x]=vis[next[Top.x]]=;
         }
         else if(next[Top.x]==n)
         {
             next[pre[pre[Top.x]]]=n;
             ans+=Top.c;
             vis[Top.x]=vis[pre[Top.x]]=;
         }
         else
         {
             ans+=Top.c;
             node New;
             New.x=++num;
             New.c=cha[pre[Top.x]]+cha[next[Top.x]]-Top.c;
             pre[New.x]=pre[pre[Top.x]],next[pre[New.x]]=New.x;
             next[New.x]=next[next[Top.x]],pre[next[New.x]]=New.x;
             vis[Top.x]=vis[next[Top.x]]=vis[pre[Top.x]]=;
             cha[num]=New.c;
             q.push(New);
         }
     }
 }

 int main()
 {
     n=in();k=in();
     for(int i=;i<=n;i++)a[i]=in();
     Pre();solve();
     printf("%lld",ans);
     return ;
 }