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);
}