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