[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

 

每次在优先队列中找到美观度最大的点并删除与它相连的点后，将它本身的权值a[i]=a[pre[i]]+a[nxt[i]]-a[i]，并再加入到优先队列中，重复操作m次。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<queue>
#define maxn 200005
using namespace std;
int n,m;
int a[maxn];
int nxt[maxn],bef[maxn];
struct data {
    int pos,v;
    bool operator <(const data tmp) const {
        return v<tmp.v;
    }
};
priority_queue<data> q;
int ans;
bool vis[maxn];
int main() {
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) {
        scanf("%d",&a[i]);
        nxt[i]=i+1;bef[i]=i-1;
        q.push((data){i,a[i]});
    }
    if(m>n/2){puts("Error!");return 0;  } 
    nxt[n]=1;bef[1]=n;
    for(int i=1;i<=m;i++) {
        data h;
        while(vis[q.top().pos]) q.pop();
        h=q.top();q.pop();
        ans+=h.v;
        a[h.pos]=a[nxt[h.pos]]+a[bef[h.pos]]-a[h.pos];
        vis[nxt[h.pos]]=vis[bef[h.pos]]=1;
        nxt[h.pos]=nxt[nxt[h.pos]];
        bef[nxt[h.pos]]=h.pos;
        bef[h.pos]=bef[bef[h.pos]];
        nxt[bef[h.pos]]=h.pos;
        q.push((data){h.pos,a[h.pos]});
    }
    printf("%d\n",ans);
}