codeforces 1283D (类似于最小生成树)

题意:有n棵圣诞树,坐标是xi(1<=i<=n),有m个人,需要把这m个人放到没有圣诞树和其他人的位置,使得所有人到离他最近的圣诞树的距离之和最小。

思路:肯定是要保证每个人离圣诞树的最小距离尽量小,所有应该是放在每个圣诞树的左右两边,(如果能放的话)。这样最小距离就是一了。然后向左右两边扩展。

这很像最小生成树,每次在优先队列里取出距离最近的点,然后把它的左右两个点放进优先队列里(如果能放的话)。

#pragma warning(disable:4996)
#include<climits>
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<map>
#include<set>
#include<queue>
#include<string>
using namespace std;
typedef long long ll;
map<ll, ll>mp;//用map来标记是否已经有了圣诞树或人
struct point
{
	int pos, val;
	bool operator <(const point &x)const
	{
		return val > x.val;//按从小到大的顺序排序
	}
};
ll ans[200005],a[200005];//记录答案
priority_queue<point>q;
int main()
{
	ll n, m, t, sum,k;
	point temp;
	scanf("%lld%lld", &n, &m);
	for (ll i = 1; i <= n; i++)
	{
		scanf("%lld", &a[i]);
		mp[a[i]]++;//标记
	}
	for (ll i = 1; i <= n; i++)
	{
		if (mp[a[i] - 1] == 0)//前一个位置没有
		{
			temp.pos = a[i] - 1;
			temp.val = 1;
			q.push(temp);//入队
			mp[temp.pos]++;//标记
		}
		if (mp[a[i] + 1] == 0)//后一个位置
		{
			temp.pos = a[i] + 1;
			temp.val = 1;
			q.push(temp);
			mp[temp.pos]++;
		}
	}
	sum = 0; k = 0;//记录答案
	while (m--)
	{
		temp = q.top(); q.pop();//取出队首
		ans[++k] = temp.pos;
		sum += temp.val;
		point Next;
		//两边的元素
		if (mp[temp.pos - 1] == 0)
		{
			Next.pos = temp.pos - 1;
			Next.val = temp.val + 1;
			q.push(Next);
			mp[Next.pos]++;
		}
		if (mp[temp.pos + 1] == 0)
		{
			Next.pos = temp.pos + 1;
			Next.val = temp.val + 1;
			q.push(Next);
			mp[Next.pos]++;
		}
	}
	printf("%lld\n", sum);
	for (ll i = 1; i <= k; i++)
	{
		printf("%lld", ans[i]);
		if (i == k)printf("\n");
		else printf(" ");
	}
	return 0;
}

 

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值