【国家集训队2011】【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

上午学校模拟赛被虐了= =下午去看了题解。不会手写堆就用set解决了0-0

谢谢COGS上CSTDIO神的题解OTZ

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
using namespace std;
int n,m;
int a[200001];
struct line
{
    int v;
    int num;
}now;
bool operator < (line a,line b)
{
    if (a.v==b.v) return a.num>b.num;
    return a.v>b.v;
}
void del(set<line> &s,int a[],int num)
{
    if (num==0) return;
    set<line>::iterator t=s.find((line){a[num],num});
    if (t!=s.end()) s.erase(t);
}
int work()
{
    int ret=0;
    int pred[200001],succ[200001];
    for (int i=1;i<=n;i++)
        pred[i]=i-1,succ[i]=i+1;
    pred[1]=n;
    succ[n]=1;
    set<line> s;
    s.clear();
    for (int i=1;i<=n;i++) s.insert((line){a[i],i});
    for (int i=1;i<=m;i++)
    {
        now=*s.begin();
        s.erase(s.begin());
        ret+=now.v;
        del(s,a,pred[now.num]);
        del(s,a,succ[now.num]);
        a[now.num]=a[pred[now.num]]+a[succ[now.num]]-a[now.num];
        pred[now.num]=pred[pred[now.num]];
        succ[now.num]=succ[succ[now.num]];
        pred[succ[now.num]]=succ[pred[now.num]]=now.num;
        s.insert((line){a[now.num],now.num});
    }
    return ret;
}
int main()
{
    freopen("nt2011_tree.in","r",stdin);
    freopen("nt2011_tree.out","w",stdout);
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    if (m>n/2)
    {
        cout<<"Error!"<<endl;
        return 0;
    }
    else
        cout<<work();
}