Advertisement
esraa_syam

^^

Oct 25th, 2023
861
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.73 KB | None | 0 0
  1. #include<bits/stdc++.h>
  2.  
  3. #define nl "\n"
  4. #define fi first
  5. #define se second
  6. #define pi 3.14159
  7. #define ll long long
  8. #define odd(a)  (a&1)
  9. #define even(a)  !(a&1)
  10. #define Mod 1'000'000'007
  11. #define INF 2'000'000'000 // 2e9
  12. #define sz(x) int(x.size())
  13. #define charToInt(s) (s - '0')
  14. #define ull unsigned long long
  15. #define number_line iota(all(vec) , 1)
  16. #define all(s) s.begin(), s.end()
  17. #define rall(v) v.rbegin() , v.rend()
  18. #define fixed(n) fixed << setprecision(n)
  19. #define Num_of_Digits(n) ((int)log10(n) + 1)
  20. #define to_decimal(bin) stoll(bin, nullptr, 2)
  21. #define Ceil(n, m) (((n) / (m)) + ((n) % (m) ? 1 : 0))
  22. #define Floor(n, m) (((n) / (m)) - ((n) % (m) ? 0 : 1))
  23. #define Upper(s) transform(all(s), s.begin(), ::toupper);
  24. #define Lower(s) transform(all(s), s.begin(), ::tolower);
  25. #define cout_map(mp) for(auto& [f, s] : mp) cout << f << "  " << s << "\n";
  26. //  ----- bits-------
  27. #define pcnt(n) __builtin_popcount(n)
  28. #define pcntll(n) __builtin_popcountll(n)
  29. #define clz(n) __builtin_clz(n)    // <---100
  30. #define clzll(n) __builtin_clzll(n)
  31. #define ctz(n) __builtin_ctz(n)  // 0001---->
  32. #define ctzll(n) __builtin_ctzll(n)
  33.  
  34. using namespace std;
  35.  
  36. template<typename T = int>
  37. istream &operator>>(istream &in, vector<T> &v) {
  38.     for (auto &x: v) in >> x;
  39.     return in;
  40. }
  41.  
  42. template<typename T = int>
  43. ostream &operator<<(ostream &out, const vector<T> &v) {
  44.     for (const T &x: v) out << x << " ";
  45.     return out;
  46. }
  47.  
  48. void esraa() {
  49.     ios_base::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
  50. #ifndef ONLINE_JUDGE
  51.     freopen("input.txt", "r", stdin), freopen("output.txt", "w", stdout);
  52. #endif
  53. }
  54.  
  55. void solve() {
  56.     int n , k;
  57.     cin >> n >> k;
  58.  
  59.     vector < vector < int > > adj(n + 5 , vector < int > ());
  60.  
  61.     for(int i = 0 ; i < n - 1 ; i++){
  62.         int u , v;
  63.         cin >> u >> v;
  64.         adj[u].push_back(v);
  65.         adj[v].push_back(u);
  66.     }
  67.  
  68.     vector < int > depth(n + 5 , 0) , siz(n + 5) , leaf;
  69.  
  70.     function < void(int , int , int) > dfs = [&](int node , int par , int d){
  71.         depth[node] = d;
  72.         siz[node] = 1;
  73.         for(auto child : adj[node]){
  74.             if(child != par){
  75.                 dfs(child , node , d + 1);
  76.                 siz[node] += siz[child];
  77.             }
  78.         }
  79.  
  80.         if(siz[node] == 1) leaf.push_back(node);
  81.     };
  82.  
  83.     dfs(1 , 0 , 1);
  84.  
  85.     vector < int > dif;
  86.  
  87.     for(int i = 0 ; i < sz(leaf) ; i++){
  88.         dif.push_back(depth[leaf[i]] - 1);
  89.     }
  90.  
  91.     sort(rall(dif));
  92.  
  93.     ll ans = 0;
  94.  
  95.     for(int i = 0 ; i < k ; i++){
  96.         ans += dif[i];
  97.     }
  98.  
  99.     cout << ans << nl;
  100.  
  101.  
  102.  
  103. }
  104.  
  105. int main() {
  106.     esraa();
  107.     int t = 1;
  108.     // cin >> t;
  109.     //cin.ignore();
  110.     while (t--)
  111.         solve();
  112.     return 0;
  113. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement