【bzoj2151】【种树】【堆+贪心】

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
题解：同机房的人去年就A了，果然我还是太弱。维护一个堆。每次从堆中去最大元素加入答案。对于取出的元素，删掉它两边的元素。再加入一个新的元素代替这三个元素。新元素的权值为取出元素左右两边的权值和减去自己的权值。

#include<iostream>
#include<cstdio>
#define p -999999999
using namespace std;
int n,m,pos[2000001],ans,h[2000001],a[2000001],l[2000001],r[2000001],u;
void up(int x)
{
    while (x>1)
     {
        if (a[h[x]]>a[h[x>>1]])
         {
           swap(h[x],h[x>>1]);
           swap(pos[h[x]],pos[h[x>>1]]);
           x=x>>1;
         }
        else break;
     }
}
void down(int x)
{
   int u;
   while (x<<1<=n)
    {
       if (a[h[x<<1]]>a[h[x<<1|1]]||x<<1==n) u=x<<1;
       else u=x<<1|1;
       if (a[h[x]]<a[h[u]])
        {
            swap(h[u],h[x]);
            swap(pos[h[u]],pos[h[x]]);
            x=u;
        }
       else break;
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    if (m>n>>1) {cout<<"Error!";return 0;}
    for (int i=1;i<=n;i++)
     {
       scanf("%d",&a[i]);
       l[i]=i-1;r[i]=i+1;pos[i]=h[i]=i;
       up(i);
     }
    l[1]=n;r[n]=1;
    for (int i=1;i<=m;i++)
     {
       u=h[1];
       ans+=a[u];a[u]=a[l[u]]+a[r[u]]-a[u];
       a[l[u]]=p;down(pos[l[u]]);
       a[r[u]]=p;down(pos[r[u]]);
       down(1);
       l[u]=l[l[u]];r[u]=r[r[u]];l[r[u]]=u;r[l[u]]=u;
     }
    printf("%d\n",ans);
}