2151: 种树

xiaoxiao2021-02-28  69

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! 题解: 贪心题, 下面引用sunny crazy的话: 首先考虑如果没有“相邻位置不能都种”这一限制会怎么样。这时就是一个裸的贪心——按照A[i]从大到小排序,然后取前M个。 那么加上限制以后会发生什么呢? 假设A[3]最大,那我们就试图去选A[3]。选中之后首先要去掉3,并且,A[2]和A[4]也都不能选了,所以将它们删掉—— 但是慢着!这可能会导致问题。假设A[3]=20,A[2]=A[4]=19,那么同时选A[2],A[4]可能比选A[3]要优!在最后的方案中可能是A[2]+A[4]而非A[3]。这种情况要怎么解决呢? 可以发现一点:由于A[3]最大,所以在最后的方案中,不可能只选A[2],A[4]中的一个。 原因很简单:假设在最优方案中选了A[2]但未选A[4],那可以简单地把A[2]换成A[3],由于未选A[4],所以这样不会产生任何矛盾,并且把A[2]换成A[3]后,总的美观度不会下降。 因此,我们先去掉2,3,4,然后加入一个新的“物品”,其权值为A[2]+A[4]-A[3],代表同时选2,4,删去3.这样,在选了3之后再选这个新物品,功效就相当于刚才所说的,把A[3]换成A[2]+A[4]。 这个新物品应该放在哪里呢?它的含义是“选2,4”,所以很容易想到,应该把它放在1,5中间。 出于方便起见,不妨在删掉2,4后直接把A[3]改成A[2]+A[4]-A[3],显然这个位置是正确的。 如此就将N个物品,需要选M个的问题转化成了在N-1个物品中选的问题。并且可以发现一个很好的性质:新的3所对应的仍然是“选中物品数+1”!(把选3换成了选2,4,即多选了一个物品) 也就是说,完全可以把新的3看做一个和1,5毫无区别的物品,现在我们只需要在1,3,5三个物品中选择M-1个!如此下去,直到选择M次,就可以得到答案。 因此描述一下算法:以A[i]为关键字建大根堆,用一个链表存放当前物品。 最初链表中元素是1~N,i的后继是i+1,前驱是i-1(当然,1的前驱是N,N的后继是1)。 执行M次操作,每一次操作都将堆顶元素k取出,ans+=A[k]。然后在链表中删除k的前驱pre和后继nxt,令A[k]=A[pre]+A[nxt]-A[k],并更新堆。 这个算法运行的很好,但你可能感觉有点虚——为什么每次选A值最大的就正确呢? 可以发现,在上面的讨论中“选3”时,我们实际上做的是声明如下事实: 在最终答案中要么选了3,要么同时选了2,4.换句话说,要么选了3,要么在此基础上选了A[2]+A[4]-A[3]。 所以我们实际上是重写了这个问题,将其变成“N-2个物品中选M-1”个的形式,如此一直化归,直到最后变成“N-2(M-1)个物品中选1个”,这时答案就是显然的。 #include<cstdio> #include<cstdlib> #include<cstring> #include<queue> using namespace std; const int N=200010; int n,m; int a[N],pre[N],nex[N]; struct node{ int num,ord; bool operator<(const node & x)const { return num<x.num; } }qp; priority_queue<node>q; bool tf[N]; void del(int x) { tf[x]=1; nex[pre[x]]=nex[x]; pre[nex[x]]=pre[x]; nex[x]=pre[x]=0; } int main() { scanf("%d%d",&n,&m); for(int i=1;i<=n;i++) { scanf("%d",&a[i]); } if(n/2<m) {printf("Error!");return 0;} for(int i=1;i<n;i++)nex[i]=i+1;nex[n]=1; for(int i=2;i<=n;i++) pre[i]=i-1;pre[1]=n; for(int i=1;i<=n;i++) q.push((node){a[i],i}); int ans=0; for(int i=1;i<=m;i++) { while(tf[q.top().ord]) q.pop(); qp=q.top();q.pop(); ans+=qp.num; int a1=pre[qp.ord],b1=nex[qp.ord]; a[qp.ord]=a[a1]+a[b1]-a[qp.ord]; del(a1);del(b1); q.push((node){a[qp.ord],qp.ord}); } printf("%d",ans); }
转载请注明原文地址: https://www.6miu.com/read-62280.html

最新回复(0)