【BZOJ1150】[CTSC2007]数据备份Backup

题意:给定n个点的坐标,要求连k条边,每个点仅可连一条边,求最小距离和。

优先队列维护相邻两点间的距离,每加一条边,将于这两点有关的三个值全删除(打标记),再将两边的值减去中间的值加入优先队列,用链表维护前后相邻的值。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <queue>
#include <cstdlib>
#include <cstring>
#define pa pair<int,int>
#define mp(a,b) make_pair(a,b)
#define N 100009
using namespace std;
int a[N];
int x[N],next[N],used[N],pre[N];
struct cmp
{
	bool operator()(const pa a,const pa b)
	{
		return a.first>b.first;
	}
};
int main()
{
	priority_queue<pa,vector<pa>,cmp> q;
	memset(used,0,sizeof(used));
	int n,k;
	cin>>n>>k;
	for (int i=1;i<=n;i++)
		cin>>a[i];
	sort(a+1,a+n+1);
	for (int i=1;i<n;i++)
		x[i]=a[i+1]-a[i],next[i]=i+1,pre[i+1]=i;
	for (int i=1;i<n;i++)
		q.push(mp(x[i],i));
	int ans=0;
	for (int i=1;i<=k;i++)
	{
		pa y=q.top();
		q.pop();
		while (used[y.second])
		{
			y=q.top();
			q.pop();
		}
		ans+=y.first;
		if (pre[y.second]==0)
		{
			used[y.second]=used[next[y.second]]=1;
			pre[next[next[y.second]]]=0;
			continue;
		}
		if (next[y.second]==n)
		{
			used[y.second]=used[pre[y.second]]=1;
			next[pre[pre[y.second]]]=n;
			continue;
		}
		q.push(mp(x[pre[y.second]]+x[next[y.second]]-y.first,y.second));
		x[y.second]=x[pre[y.second]]+x[next[y.second]]-y.first;
		used[pre[y.second]]=1;
		used[next[y.second]]=1;
		pre[y.second]=pre[pre[y.second]];
		next[y.second]=next[next[y.second]];
		next[pre[y.second]]=y.second;
		pre[next[y.second]]=y.second;
	}
	cout<<ans<<endl;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值