洛谷1484 BZOJ2151 种树 贪心+堆

题目链接
洛谷题意：
给你一个长度为n的序列，每个位置有一个权值，相邻位置不能同时被选，问你选k个数得到的最大权值是多少（可以选的数不足k个）。

BZOJ题意：把洛谷的序列改成环，并且要判断无解情况，但是必须选足k个数。

题目差不多。
这是一个比较经典的题目，这题不难想出可以n^2dp，但是好像没法优化。我们考虑贪心，记录每个点左侧和右侧的点是哪个点，然后记录每个点的点权和每个点是否被考虑过，把当前的所有点放入堆，每次取堆顶没有被选过的第一个点，把它拿出来加到答案中，标记左侧和右侧点被当下考虑过，并且把它左右两个点的和减去原来的的权值当作这个点的新的权值，左侧看作原来左侧的左侧，右侧看作原来右侧的右侧，这样就可以看作形成了一个新点，把新点再加入堆中，我们发现其实取出新点其实是去掉一个点加进两个点，也是相当于多了一个点，于是只需要取k次堆顶就可以了。
感觉如果没看明白的话看看代码也就明白是在干嘛了。代码里注释了洛谷和BZOJ题目有区别的地方，代码是在BZOJ直接能过的代码。

#include <bits/stdc++.h>
using namespace std;

int n,k,l[500010],r[500010],book[500010];
long long a[500010],ans;
priority_queue<pair<long long,int> > q;
int main()
{
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;++i)
    scanf("%lld",&a[i]);
    if(k>(n>>1))//BZOJ与洛谷的差别 
    {
        printf("Error!\n");
        return 0;
    } 
    for(int i=1;i<=n;++i)
    {
        l[i]=i-1;
        r[i]=i+1;
        q.push(make_pair(a[i],i));
    }
    l[1]=n;//BZOJ与洛谷的差别 
    r[n]=1;
    while(k--)
    {
        int x=q.top().second;
        long long val=q.top().first;
        q.pop();        
        while(book[x])
        {
            x=q.top().second;
            val=q.top().first;
            q.pop();
        }
//      if(val<0)//BZOJ与洛谷的差别 
//      break;
        ans+=val;
        book[l[x]]=1;
        book[r[x]]=1;
        a[x]=a[l[x]]+a[r[x]]-a[x];
        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("%lld\n",ans);
    return 0;
}