Cod sursa(job #2715183)

Utilizator sichetpaulSichet Paul sichetpaul Data 3 martie 2021 10:28:23
Problema Salvare Scor 30
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.33 kb
#include <bits/stdc++.h>
#define Nmax 1002
using namespace std;

ifstream fin("salvare.in");
ofstream fout("salvare.out");

int N, K, nr;
int dp[Nmax];
vector<int> G[Nmax], towers, curr;

void DFS(int node, int father, int val) {
    bool sons = 0;
    dp[node] = 0;
    for (auto it: G[node]) {
        if (it == father) continue;
        sons = 1;
        DFS(it, node, val);
        dp[node] = max(dp[it], dp[node]);
    }

    if (sons) {
        dp[node]++;
        if (dp[node] == val) dp[node] = 0, ++nr, curr.push_back(node);
    }
}
bool check(int val) {
    for (int i = 1; i <= N; ++i) {
        nr = 0;
        curr.clear();
        DFS(i, 0, val);

        if (nr > K) continue;
        towers.clear();
        for (auto it: curr)
            towers.push_back(it);
        return 1;
    }
       return 0;
}
int main()
{
    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);
    }

    int le = 1, ri = N, ans = N;
    while (le <= ri) {
        int mid = (le + ri) / 2;
        if (check(mid)) ans = mid, ri = mid - 1;
        else le = mid + 1;
    }

    fout << ans << '\n';
    sort(towers.begin(), towers.end());
    for (auto it: towers)
         fout << it << " ";

    return 0;
}