题目链接: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;
}