3493441984zz
2019-02-15 14:00:41
题解好像没有图,发个有图的,希望能帮助你们更好的理解
其实就是一个贪心吧,见的挺多的,一开始做的话很难想到这个反悔机制
我们把每相邻两个办公楼之间的距离算出来,抽象出来点,也就是有
那么,当我们选了第
那么这样做了之后,题目就变为,给定
那么问题转化了之后呢???
我们弄一个小根堆(本蒟蒻用的优先队列代替)
把每一个点放进去,然后每次贪心取最小值,取了一个点后就把相邻的点标记为不能取
如果你认为这是对的话,就看看下面这个样例(楼层
长方形是楼层,圆形是抽象出来的点,那么如果按照上面的思路的话,我们会先取
那么我们就只能取
我们用肉眼都能看出来选择
当我们取了一个点后,我们要再加入一个点,这个点的权值为左边的点权
我们边模拟边解释:
当我们取了
如下图:
那么优先队列下一个点为
最后如下图:
看出来了吗?其实这就相当于我们取了
也就是说取了
你看懂了这个反悔机制吗
至于左右节点的更新,我们用双向链表实现:
接下来是美滋滋的代码时间~~~
#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
#define N 500007
#define inf 0x3f3f3f3f
using namespace std;
struct Place
{
int val,l,r;
}p[N];
struct Node
{
int val,id;
bool operator <(Node it) const
{
return val>it.val;
}
};
int n,m,ans,last;
bool vis[N];
priority_queue<Node> q;
void Del(int x)
{
p[x].l=p[p[x].l].l;
p[x].r=p[p[x].r].r;
p[p[x].l].r=x;
p[p[x].r].l=x;
}
int main()
{
scanf("%d%d%d",&n,&m,&last);
for(int i=1;i<n;++i)
{
int in;
scanf("%d",&in);
p[i].val=in-last;
last=in;
p[i].l=i-1;
p[i].r=i+1;
q.push((Node){p[i].val,i});
}
p[0].val=p[n].val=inf;//注意这里
for(int i=1;i<=m;++i)
{
while(vis[q.top().id])
q.pop();
Node now=q.top();
q.pop();
ans+=now.val;
vis[p[now.id].l]=vis[p[now.id].r]=1;
p[now.id].val=p[p[now.id].l].val+p[p[now.id].r].val-p[now.id].val;
q.push((Node){p[now.id].val,now.id});
Del(now.id);
}
printf("%d",ans);
return 0;
}