bzoj2151 种树 双向链表+堆

2151: 种树

Time Limit: 10 Sec  Memory Limit: 259 MB
Submit: 1151  Solved: 613
[Submit][Status][Discuss]

Description

A城市有一个巨大的圆形广场，为了绿化环境和净化空气，市政府决定沿圆形广场外圈种一圈树。园林部门得到指令后，初步规划出n个种树的位置，顺时针编号1到n。并且每个位置都有一个美观度Ai，如果在这里种树就可以得到这Ai的美观度。但由于A城市土壤肥力欠佳，两棵树决不能种在相邻的位置（i号位置和i+1号位置叫相邻位置。值得注意的是1号和n号也算相邻位置！）。最终市政府给园林部门提供了m棵树苗并要求全部种上，请你帮忙设计种树方案使得美观度总和最大。如果无法将m棵树苗全部种上，给出无解信息。

Input

输入的第一行包含两个正整数n、m。第二行n个整数Ai。

Output

输出一个整数，表示最佳植树方案可以得到的美观度。如果无解输出“Error!”，不包含引号。

Sample Input

【样例输入1】
7 3
1 2 3 4 5 6 7
【样例输入2】
7 4
1 2 3 4 5 6 7

Sample Output

【样例输出1】
15

【样例输出2】
Error!
【数据规模】
对于全部数据：m<=n；
-1000<=Ai<=1000
N的大小对于不同数据有所不同：
数据编号 N的大小 数据编号 N的大小
1 30 11 200
2 35 12 2007
3 40 13 2008
4 45 14 2009
5 50 15 2010
6 55 16 2011
7 60 17 2012
8 65 18 199999
9 200 19 199999
10 200 20 200000

HINT

Source

把整个环用双向链表串起来

每次选个最大的x

用a[nxt[x]]+a[pre[x]]-a[x]替代它，删去pre[x],nxt[x]

 #pragma GCC optimize(2)
 #pragma G++ optimize(2)
 #include<iostream>
 #include<algorithm>
 #include<cmath>
 #include<cstdio>
 #include<cstring>
 #include<queue>

 #define fzy pair<int,int>
 #define N 200007
 using namespace std;
 inline int read()
 {
     int x=,f=;char ch=getchar();
     while(!isdigit(ch)){if(ch=='-')f=-;ch=getchar();}
     while(isdigit(ch)){x=(x<<)+(x<<)+ch-'';ch=getchar();}
     return x*f;
 }

 int ans;
 int n,m,cnt;
 int a[N],pre[N],nxt[N];
 bool flag[N];
 priority_queue<fzy,vector<fzy> >q;

 void del(int p)
 {
     flag[p]=true;
     int l=pre[p],r=nxt[p];
     nxt[p]=pre[p]=;
     nxt[l]=r,pre[r]=l;
 }
 void solve()
 {
     while(flag[q.top().second])q.pop();
     int x=q.top().second;ans+=a[x];q.pop();
     a[x]=a[pre[x]]+a[nxt[x]]-a[x];
     del(pre[x]),del(nxt[x]);
     q.push(make_pair(a[x],x));
 }
 int main()
 {
     n=read(),m=read();
     for (int i=;i<=n;i++)a[i]=read();
     if(m>n/){puts("Error!");return ;}
     for (int i=;i<=n;i++)pre[i]=i-;pre[]=n;
     for (int i=;i<=n;i++)nxt[i]=i+;nxt[n]=;
     for (int i=;i<=n;i++)
         q.push(make_pair(a[i],i));
     for (int i=;i<=m;i++)solve();
     printf("%d\n",ans);
 }