【国家集训队2011】【BZOJ2151】种树

最新推荐文章于 2023-07-04 21:18:59 发布

原创最新推荐文章于 2023-07-04 21:18:59 发布 · 1.3k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#堆

本文介绍了一个关于在圆形广场种植树木以最大化美观度的算法问题。该问题需要在给定的条件下选择最佳的植树位置，避免相邻种植，并实现美观度总和的最大化。文章通过使用set数据结构提供了一种解决方案。

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();
}