zyf2000

"No regrets."

[BZOJ2151]种树(贪心+堆)

题目描述

传送门

题解

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

代码

#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…

阅读更多
版权声明:转载请注明出处:http://blog.csdn.net/clove_unique https://blog.csdn.net/Clove_unique/article/details/52346629
个人分类: 题解 贪心
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

不良信息举报

[BZOJ2151]种树(贪心+堆)

最多只允许输入30个字

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭