Christmas Trees

思路:
如下 ，从三个小点开始 ，像不像bfs ，没错就是bfs ，拿个map来记录每个点是否有树或者人， 把可以放的位置放进答案容器里就行了。

#include<iostream>
#include<cstring>
#include<vector>
#include<algorithm>
#include<queue>
#include<map>
using namespace std;
typedef long long ll;
const int N=200010;
const int mod = 998244353;
map<int,bool>mp;
vector<int>res;
int a[N];
typedef pair<int,int> pii;
queue<pii>Q;
int main()
{	
	int n,m;
	cin>>n>>m;
	int cnt=0;
	for(int i=0;i<n;i++)
	{
		cin>>a[i];
		mp[a[i]]=true;
		Q.push({1,a[i]-1});
		Q.push({1,a[i]+1});
	}
	ll rr=0;
	int maxv=0;
	while(Q.size())
	{
		auto t =Q.front();
		Q.pop();
		int loc = t.second;
		int dis = t.first;
		if(!mp[loc])
		{
			m--;
			res.push_back(loc);
			mp[loc]=true;
			rr += dis;
			if(!m)	break;
			Q.push({dis+1,loc+1});
			Q.push({dis+1,loc-1});
		}
	}
	cout<<rr<<endl;
	for(int i=0;i<res.size();i++)
		cout<<res[i]<<' ';
	cout<<endl;
	return 0;

}

下面发一份超时代码：

#include<iostream>
#include<cstring>
#include<vector>
#include<algorithm>
#include<queue>
#include<map>
using namespace std;
typedef long long ll;
const int N=200010;
const int mod = 998244353;
map<int,bool>mp;
vector<int>res[N];
int a[N];
int main()
{	
	int n,m;
	cin>>n>>m;
	int cnt=0;
	for(int i=0;i<n;i++)
	{
		cin>>a[i];
		mp[a[i]]=true;
	}
	ll rr=0;
	int maxv=0;
	for(int t=1;;t++)
	{
			for(int j=0;j<n;j++)
			{
				if(!mp.count(a[j]-t))
			{
				res[t].push_back(a[j]-t);
				mp[a[j]-t]=true;
				cnt++;
				rr+=t;
			}
				if(cnt>=m)	break;
				if(!mp.count(a[j]+t))
					{
						res[t].push_back(a[j]+t);
						mp[a[j]+t]=true;
						cnt++;
						rr+=t;
					}
				if(cnt>=m)	break;	

			}
			if(cnt>=m)	{
				maxv=t;
				break;
					}
	}
	cout<<rr<<endl;
	for(int i=1;i<=maxv;i++)
		for(int j=0;j<res[i].size();j++)
		cout<<res[i][j]<<' ';
		cout<<endl; 
	return 0;

}

这样确实慢 ，做了好多无用功。遍历了好多遍数组，真有你的。