一道经典反悔贪心模板题,但是 k k k只有 5000 5000 5000,所以题解变成了 k 2 k^2 k2的优化 d p dp dp.
题意
你的宇宙飞船要会见 k k k位领导人,只有一些固定时间能让领导人上下车,但是值得高兴的是领导人只会待一会.你不能同时会见两个以上的领导人,同一个时间只能让一个领导人上车或者下车.问会见这些领导人至少要多少时间.
题解
转化模型.对时间排序后计算相邻差,则问题转化为取
k
k
k个数使得没有两个相邻,此为经典反悔贪心模板.
首先构建双向链表,取走最大的数的时候,由于可能存在两边之和大于它本身的情况,我们删去两边的结果,将
a
i
+
1
+
a
i
−
1
−
a
i
a_{i+1}+a_{i-1}-a_i
ai+1+ai−1−ai赋值给
a
i
a_i
ai,这样如果再取最大值,取到
a
i
a_i
ai的时候,答案就相当于取了两边的数字.
这样的贪心进行
k
k
k次便可得到最优解.
谢谢大家.
const int yuzu=5e5;
const ll inf=1e18;
typedef ll fuko[yuzu|10];
typedef pair<ll,int> pli;
fuko net,pre,a;
int main() {
int n,k,i;
read(k),read(n);
for (i=1;i<=n;++i) read(a[i]);
sort(a+1,a+n+1);
for (i=1;i<n;++i) {
a[i]=a[i+1]-a[i];
net[i]=i+1,pre[i]=i-1;
}
net[n-1]=0;
ll zw=0;
priority_queue<pli,vector<pli>,greater<pli>> q;
for (i=1;i<n;++i) q.push(pli(a[i],i));
for (;k--;) {
for (;q.top().first^a[q.top().second];) q.pop();
int u=q.top().second,l=pre[u],r=net[u];
zw+=a[u]; q.pop();
pre[net[u]=net[r]]=u;
net[pre[u]=pre[l]]=u;
a[u]=l&&r?min(inf,a[l]+a[r]-a[u]):inf;
a[l]=a[r]=inf;
q.push(pli(a[u],u));
}
printf("%lld\n",zw);
}