Cod sursa(job #1243660)

Utilizator CosminRusuCosmin Rusu CosminRusu Data 16 octombrie 2014 10:18:53
Problema Salvare Scor 65
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.6 kb
#include <fstream>
#include <iostream>
#include <vector>
#include <bitset>
#include <string.h>
#include <algorithm>
#include <iomanip>
#include <math.h>
#include <time.h>
#include <stdlib.h>
#include <set>
#include <map>
#include <string>
#include <queue>
#include <deque>

using namespace std;

const char infile[] = "salvare.in";
const char outfile[] = "salvare.out";

ifstream fin(infile);
ofstream fout(outfile);

const int MAXN = 100005;
const int oo = 0x3f3f3f3f;

typedef vector<int> Graph[MAXN];
typedef vector<int> :: iterator It;

const inline int min(const int &a, const int &b) { if( a > b ) return b;   return a; }
const inline int max(const int &a, const int &b) { if( a < b ) return b;   return a; }
const inline void Get_min(int &a, const int b)    { if( a > b ) a = b; }
const inline void Get_max(int &a, const int b)    { if( a < b ) a = b; }

int dp[MAXN], deg[MAXN];
Graph g;
int N, K;
queue <int> Q;
vector <int> act, sol;

inline bool Okay(int M) {
    vector <int>().swap(act);
    for(int i = 1 ; i <= N ; ++ i) {
        deg[i] = g[i].size();
        dp[i] = oo;
        if(deg[i] == 1) {
            Q.push(i);
            dp[i] = M;
        }
    }
    while(!Q.empty()) {
        int node = Q.front();
        Q.pop();
        deg[node] = 0;
        if(dp[node] == 0) {
            act.push_back(node);
            if(act.size() > K) {
                queue <int>().swap(Q);
                return false;
            }
            dp[node] = (M << 1) + 1;
        }
        for(It it = g[node].begin(), fin = g[node].end(); it != fin ; ++ it) {
            -- deg[*it];
            dp[*it] = min(dp[node] - 1, dp[*it]);
            if(deg[*it] == 1)
                Q.push(*it);
        }
    }
    return true;
}

inline int getAns() {
    int li = 0, ls = N, ret = -1;
    while(li <= ls) {
        int mid = ((li + ls) >> 1);
        if(Okay(mid)) {
            ls = mid - 1;
            ret = mid;
            sol = act;
        }
        else
            li = mid + 1;
    }
    return ret;
}

int main() {
    cin.sync_with_stdio(false);
    #ifndef ONLINE_JUDGE
    freopen(infile, "r", stdin);
    freopen(outfile, "w", stdout);
    #endif
    fin >> N >> K;
    for(int i = 1 ; i < N ; ++ i) {
        int x, y;
        fin >> x >> y;
        g[x].push_back(y);
        g[y].push_back(x);
    }
    memset(deg, 0, sizeof(deg));
    fout << getAns() << '\n';
    for(int i = 0 ; i < sol.size() ; ++ i)
        deg[sol[i]] = 1;
    for(int i = 1 ; i <= N && sol.size() != K ; ++ i)
        if(!deg[i])
            sol.push_back(i);
    for(auto it:sol)
        fout << it << ' ';
    fin.close();
    fout.close();
    return 0;
}