题意:有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;
}