Christmas Trees(BFS好题)

Problem - 1283D - Codeforces

在一条无限的数线上有n棵圣诞树。第i棵树在位置xi上生长。所有的xi都被保证是独立的。

每个整数点要么被圣诞树占据，要么被人类占据，要么根本不被占据。非整数点不能被任何东西占据。

有m个想庆祝圣诞节的人。让y1,y2,...,ym是人们的位置（注意所有的值x1,x2,...,xn,y1,y2,...,ym应该是不同的，所有yj应该是整数）。你想找到这样一种人的排列方式，使得∑j=1mmini=1n|xi-yj|的值是最小的（换句话说，所有的人到最近的圣诞树的距离之和应该是最小的）。

换句话说，让dj是第j个人类到最近的圣诞树的距离（dj=mini=1n|yj-xi|）。然后你需要选择这样的位置y1,y2,...,ym，使∑j=1mdj尽可能的小。

输入
输入的第一行包含两个整数n和m（1≤n,m≤2⋅105）--圣诞树的数量和人的数量。

输入的第二行包含n个整数x1,x2,...,xn（-109≤xi≤109），其中xi是第i棵圣诞树的位置。保证所有的xi都是不同的。

输出
在第一行打印一个整数res--∑j=1mmini=1n|xi-yj|的最小可能值（换句话说，所有的人到最近的圣诞树的距离之和）。

第二行打印m个整数y1,y2,...,ym（-2⋅109≤yj≤2⋅109），其中yj是第j个人的位置。所有的yj应该是不同的，所有的值x1,x2,...,xn,y1,y2,...,ym应该是不同的。

如果有多个答案，则打印其中任何一个。

例子
inputCopy
2 6
1 5
outputCopy
8
-1 2 6 4 0 3 
输入复制
3 5
0 3 1
输出拷贝
7
5 -2 4 -1 2 
 

题意:n棵固定位置的树,往里面插m个人,要求每个人与最近距离的树距离之和最小

思路:把每棵位置-1,+1的树放到队列中bfs

#include<iostream>
#include<map>
#include<queue>
using namespace std;
typedef pair<int,int> PII;
int n,m;
queue<PII> q;
map<int,int> f;
queue<int> k;
void bfs()
{
	long long ans = 0;
	while(q.size())
	{
		PII now = q.front();
		q.pop();
		if(!f[now.second])
		{
			m--;
			ans += now.first;
			k.push(now.second);
			if(m <= 0)
			break;
			f[now.second] = 1;
			if(!f[now.second + 1])
			q.push({now.first+1,now.second+1});
			if(!f[now.second - 1])
			q.push({now.first+1,now.second-1});
			
		}
	}
	printf("%lld\n",ans);
	while(k.size())
	{
		int p = k.front();
		k.pop();
		printf("%d ",p);
		
	}
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i = 1; i <= n; i++)
	{
		int x;
		scanf("%d",&x);
		f[x] = 1;
		q.push({1,x-1});
		q.push({1,x+1});
	}
	bfs();
}