【题解】CF1283D Christmas Trees

Sol

首先，显然要先放距离为 $1$ 的位置，再放 $2$ 的，以此类推。
因为是最短距离，考虑 bfs。
一开始把圣诞树全部丢到队列里，然后用这些圣诞树来扩展距离为 $1$ 的，再用距离为 $1$ 的去扩展距离为 $2$ 的。
然后开个 map 防止一个位置被多次扩展。
当扩展到的点的个数为 $m$ 时停止扩展。
时间复杂度 $O(n+m)$

Code

//LYC_music yyds!
#include<bits/stdc++.h>
#define IOS ios::sync_with_stdio(0)
#define lowbit(x) (x&(-x))
#define int long long
using namespace std;
inline char gc()
{
	static char buf[1000000],*p1=buf,*p2=buf;
	return p1==p2&&(p2=(p1=buf)+fread(buf,1,1000000,stdin),p1==p2)?EOF:*p1++;
}
int read()
{
	int pos=1,num=0;
	char ch=getchar();
	while (!isdigit(ch))
	{
		if (ch=='-') pos=-1;
		ch=getchar();
	}
	while (isdigit(ch))
	{
		num=num*10+(int)(ch-'0');
		ch=getchar();
	}
	return pos*num;
}
void write(int x)
{
	if (x<0)
	{
		putchar('-');
		write(-x);
		return;
	}
	if (x>=10) write(x/10);
	putchar(x%10+'0');
}
void writesp(int x)
{
	write(x);
	putchar(' ');
}
void writeln(int x)
{
	write(x);
	putchar('\n');
}
const int N=2e5+10;
int n,m,sum;
const int dx[]={-1,1};
queue<int> q;
map<int,int> dis;
vector<int> ans;
void bfs()
{
	while (ans.size()<m)
	{
		int u=q.front(); q.pop();
		sum+=dis[u];
		if (dis[u]) ans.push_back(u);
		for (int i=0;i<2;i++)
		{
			int xx=u+dx[i];
			if (dis.count(xx)) continue;
			dis[xx]=dis[u]+1; q.push(xx);
		}
	}
}
signed main()
{
	n=read(); m=read();
	for (int i=1;i<=n;i++)
	{
		int x=read();
		q.push(x); dis[x]=0;
	}
	bfs();
	writeln(sum);
	for (auto x:ans)
		writesp(x);
}