BZOJ 2151 种树

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

---

【题目分析】
可修正的贪心。

---

【代码】

#include <queue>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
struct node{
    int pos;
    int val;
}testa,testb;
struct cmp
{
    bool operator () (const node & a,const node & b)
    {return a.val<b.val;}
};
priority_queue <node,vector<node>,cmp> q;
int n,m,ans=0;
int pre[200001],nxt[200001],a[200001],del[200001];
void de(int x)
{
    del[x]=1;
    nxt[pre[x]]=nxt[x];
    pre[nxt[x]]=pre[x];
    nxt[x]=pre[x]=0;
}
void add()
{
    while (del[q.top().pos]) q.pop();
    int x=q.top().pos;ans+=a[x];
//  printf("%d %d %d\n",x,pre[x],nxt[x]);
    q.pop();
    a[x]=a[pre[x]]+a[nxt[x]]-a[x];
    de(pre[x]); de(nxt[x]);
    node xx;
    xx.pos=x; xx.val=a[x];
    q.push(xx);
}
int main()
{
    scanf("%d%d",&n,&m);
    if (m>n/2) { printf("Error!\n"); return 0; }
    for (int i=1;i<=n;++i)
    {scanf("%d",&a[i]);}
    for (int i=1;i<n;++i) nxt[i]=i+1; nxt[n]=1;
    for (int i=2;i<=n;++i)pre[i]=i-1; pre[1]=n;
    for (int i=1;i<=n;++i)
    {
        node xx;
        xx.pos=i;
        xx.val=a[i];
        q.push(xx);
    }
    for (int i=1;i<=m;++i) add();
    printf("%d\n",ans);
}