Cod sursa(job #1598631)

Utilizator touristGennady Korotkevich tourist Data 13 februarie 2016 02:58:53
Problema Salvare Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.52 kb
#include <bits/stdc++.h>
#define Nmax 1005
#define pb push_back
#define INF 1000000000

using namespace std;

vector <int> L[Nmax],vec,ans;
int Best[Nmax],Cnt,Info[Nmax];
bool used[Nmax];

inline void Dfs(int nod, int tata, int x)
{
    int maxx=-1,minn=INF;
    for(auto it : L[nod])
    {
        if(it==tata) continue;
        Dfs(it,nod,x);
        if(!Info[it]) maxx=max(maxx,Best[it]);
        else minn=min(minn,Best[it]);
    }
    if(maxx+1 <= abs(minn+1) && minn<0)
    {
        Best[nod]=minn+1;
        Info[nod]=1;
    }
    else
        if(maxx+1<x)
        {
            Best[nod]=maxx+1;
            Info[nod]=0;
        }
        else
        {
            vec.pb(nod); Best[nod]=-x;
            Info[nod]=1;
        }

}

int main()
{
    int i,x,y,k,n;
    ifstream cin("salvare.in");
    ofstream cout("salvare.out");
    cin>>n>>k;
    for(i=1;i<n;++i)
    {
        cin>>x>>y;
        L[x].pb(y); L[y].pb(x);
    }
    int st=0,dr=n,mij,sol;
    while(st<=dr)
    {
        mij=((st+dr)>>1);
        vec.clear(); Dfs(1,0,mij);
        if(!Info[1]) vec.pb(1);
        if(vec.size()<=k)
        {
            sol=mij; dr=mij-1;
            ans=vec;
        }
        else st=mij+1;
    }
    cout<<sol<<"\n";
    for(auto it : ans) used[it]=1;
    int nr=ans.size();
    for(i=1;i<=n && nr<k;++i)
        if(!used[i])
        {
            used[i]=1; ++nr;
        }
    for(i=1;i<=n;++i)
        if(used[i]) cout<<i<<" ";
    cout<<"\n";
    return 0;
}