Cod sursa(job #2715214)

Utilizator sichetpaulSichet Paul sichetpaul Data 3 martie 2021 11:40:13
Problema Salvare Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.99 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];

void DFS(int node, int father, int val, int dist) {
    int len = dist;
    if (dist == val + 1) len = 1, ++nr;
    else ++len;
    for (auto it: G[node])
        if (it != father) DFS(it, node, val, len);
}
bool check(int val) {
    for (int i = 1; i <= N; ++i) {
        nr = 1;
        DFS(i, 0, val, 0);
        if (val == 1 && nr <= K) cout << i << '\n';
        if (nr <= K) 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';

    return 0;
}