bzoj2151[国家集训队2011] 种树

题目链接:bzoj2151
题目大意:
在一个有n个位置的环中,选出m个互不相邻的位置,使选出位置上的值的和最大。n<=200000

题解:
贪心+优先队列
不知道怎么处理那个限制qwq哎好菜啊于是膜了发题解
[转自http://blog.sina.com.cn/s/blog_c5566b0f0102v7ns.html]

首先考虑如果没有“相邻位置不能都种”这一限制会怎么样。这时就是一个裸的贪心——按照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个”,这时答案就是显然的。

题解写得很详细。看完就懂了。嗯。
于是就到了自己乱搞时间。
其实打得有点乱,所以交的时候很虚啊。
没想到一A了。高兴。
不怎么会手打堆,所以偷懒直接上了priority_queue。
感觉主要是处理修改后的前驱后继比较容易打错吧,要先想清楚。

#include<cstdio>
#include<cstdlib>
#include<queue>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long LL;
#define maxn 201000

struct node
{
    int x,id;
    bool operator < (const node y) const
    {
        if (x!=y.x) return x<y.x;
        return id<y.id;
    }
};
bool bo[maxn];
int a[maxn],pre[maxn],next[maxn];
priority_queue<node> q;
int main()
{
    //freopen("a.in","r",stdin);
    //freopen("a.out","w",stdout);
    int n,m,i,x,ans;
    scanf("%d%d",&n,&m);
    for (i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        node tt;tt.id=i;tt.x=a[i];
        pre[i]=i-1;next[i]=i+1;
        if (i==1) pre[i]=n;
        if (i==n) next[i]=1;
        q.push(tt);bo[i]=false;
    }
    if (n<m*2) printf("Error!\n");
    else
    {
        ans=0;
        while (m--)
        {
            node now;
            while (1)
            {
                now=q.top();q.pop();
                if (!bo[now.id]) break;
            }
            x=now.id;
            ans+=now.x;
            bo[pre[x]]=bo[next[x]]=true;
            int p=pre[x],t=next[x];
            a[x]=now.x=a[p]+a[t]-a[x];
            pre[x]=pre[p];next[x]=next[t];
            next[pre[p]]=x;pre[next[t]]=x;
            q.push(now);
        }
        printf("%d\n",ans);
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值