bzoj2151 种树 双向链表+堆
2151: 种树
Time Limit: 10 Sec Memory Limit: 259 MBSubmit: 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
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
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]
1 #pragma GCC optimize(2) 2 #pragma G++ optimize(2) 3 #include<iostream> 4 #include<algorithm> 5 #include<cmath> 6 #include<cstdio> 7 #include<cstring> 8 #include<queue> 9 10 #define fzy pair<int,int> 11 #define N 200007 12 using namespace std; 13 inline int read() 14 { 15 int x=0,f=1;char ch=getchar(); 16 while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();} 17 while(isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();} 18 return x*f; 19 } 20 21 int ans; 22 int n,m,cnt; 23 int a[N],pre[N],nxt[N]; 24 bool flag[N]; 25 priority_queue<fzy,vector<fzy> >q; 26 27 void del(int p) 28 { 29 flag[p]=true; 30 int l=pre[p],r=nxt[p]; 31 nxt[p]=pre[p]=0; 32 nxt[l]=r,pre[r]=l; 33 } 34 void solve() 35 { 36 while(flag[q.top().second])q.pop(); 37 int x=q.top().second;ans+=a[x];q.pop(); 38 a[x]=a[pre[x]]+a[nxt[x]]-a[x]; 39 del(pre[x]),del(nxt[x]); 40 q.push(make_pair(a[x],x)); 41 } 42 int main() 43 { 44 n=read(),m=read(); 45 for (int i=1;i<=n;i++)a[i]=read(); 46 if(m>n/2){puts("Error!");return 0;} 47 for (int i=1;i<=n;i++)pre[i]=i-1;pre[1]=n; 48 for (int i=1;i<=n;i++)nxt[i]=i+1;nxt[n]=1; 49 for (int i=1;i<=n;i++) 50 q.push(make_pair(a[i],i)); 51 for (int i=1;i<=m;i++)solve(); 52 printf("%d\n",ans); 53 }