[BZOJ2151]种树（贪心+堆）

题目描述

传送门

题解

$n^3$的dp是很容易想到的，把最大值优化一下可以做到$n^2$。
首先考虑没有相邻的两个不能同时种的约束，可以发现就是一个简单的贪心。但是如果加上这个约束是贪心是不行的，反例也很简单。
可以借鉴网络流加反向弧的思想，就是跑不对是可以退流的。那么我们可以还像贪心一样搞，每一次找出最大的点，然后将这个点的左右两个点删去，将这个点的值改为它左右的点的和减去它原来的权值。

代码

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
using namespace std;
#define N 200005
#define pii pair<int,int>

priority_queue <pii,vector<pii> > q;
int n,m,ans;
int a[N],L[N],R[N];
bool flag[N];

int main()
{
    scanf("%d%d",&n,&m);
    if (m>(n>>1))
    {
        puts("Error!");
        return 0;
    }
    for (int i=1;i<=n;++i)
    {
        scanf("%d",&a[i]);
        L[i]=i-1; R[i]=i+1;
        q.push(make_pair(a[i],i));
    }
    L[1]=n; R[n]=1;
    ans=0;
    while (m--)
    {
        int val=q.top().first,x=q.top().second; q.pop();
        while (flag[x]) val=q.top().first,x=q.top().second,q.pop();
        ans+=val;
        a[x]=a[L[x]]+a[R[x]]-a[x];
        flag[L[x]]=true; flag[R[x]]=true;
        L[x]=L[L[x]];
        R[x]=R[R[x]];
        R[L[x]]=x;
        L[R[x]]=x;
        q.push(make_pair(a[x],x));
    }
    printf("%d\n",ans);
}

总结

1、考虑负权！goudie…