codeforces 1283D（BFS）

题意描述

有n个人，要求找到m个圣诞树，使n个人到m个圣诞树之间的距离之和最小

思路

使用bfs来做，贪心的考虑，我们先找与n个点距离为1的圣诞树，然后再找距离为2的圣诞树，依次类推。

AC代码

#include<bits/stdc++.h>
#define x first
#define y second
#define PB push_back
#define mst(x,a) memset(x,a,sizeof(x))
#define all(a) begin(a),end(a)
#define rep(x,l,u) for(ll x=l;x<u;x++)
#define rrep(x,l,u) for(ll x=l;x>=u;x--)
#define sz(x) x.size()
#define ins(x) inserter(x,x.begin())
#define IOS ios::sync_with_stdio(false);cin.tie(0);
using namespace std;
typedef unsigned long long ull;
typedef pair<int,int> PII;
typedef pair<long,long> PLL;
typedef pair<char,char> PCC;
typedef long long ll;
const int N=4*1e5+10;
const int M=1e6+10;
const int INF=0x3f3f3f3f;
const int MOD=1e9+7;
void solve(){
    queue<PII> q;
    vector<int> v;
    ll ans=0;
    map<int,int> used;
    int n,m;cin>>n>>m;
    rep(i,0,n){
        int x;cin>>x;
        used[x]=1;
        q.push({x+1,1});
        q.push({x-1,1});
    }
    while(m>0 && q.size()){
        PII cur=q.front();q.pop();
        if(!used[cur.x]){
            ans+=cur.y;
            used[cur.x]=1;
            m--;
            v.PB(cur.x);
        }
        if(!used[cur.x+1]) q.push({cur.x+1,cur.y+1});
        if(!used[cur.x-1]) q.push({cur.x-1,cur.y+1});
    }
    cout<<ans<<endl;
    rep(i,0,v.size()) cout<<v[i]<<' ';
    cout<<endl;
}
int main(){
    IOS;
    //int t;cin>>t;
    //while(t--){
        solve();
    //}
    return 0;
}