Ankit_132

E

Dec 27th, 2023
490
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.22 KB | None | 0 0
  1.  
  2. #include <bits/stdc++.h>
  3.  
  4. using namespace std;
  5.  
  6. #define _test   int _TEST; cin>>_TEST; while(_TEST--)
  7. #define pb     push_back
  8.  
  9. int main()
  10. {
  11.     _test
  12.     {
  13.         int n;
  14.         cin>>n;
  15.  
  16.         vector<vector<int>> tree(n);
  17.  
  18.         for(int i=0; i<n-1; i++)
  19.         {
  20.             int u, v;
  21.             cin>>u>>v;
  22.  
  23.             u--, v--;
  24.  
  25.             tree[u].pb(v);
  26.             tree[v].pb(u);
  27.         }
  28.  
  29.         vector<int> cnt(n);
  30.         vector<int> dis(n, 1e9);
  31.         vector<int> ht(n, 0);
  32.  
  33.         function<void(int, int)> DFS = [&](int u, int p)
  34.         {
  35.             int f = 1;
  36.  
  37.             for(auto v: tree[u])
  38.             {
  39.                 if(v == p)      continue;
  40.  
  41.                 f = 0;
  42.                 ht[v] = ht[u]+1;
  43.                 DFS(v, u);
  44.                 dis[u] = min(dis[u], dis[v]+1);
  45.                 cnt[u] += cnt[v];
  46.             }
  47.  
  48.             if(f)
  49.             {
  50.                 cnt[u] = 1;
  51.                 dis[u] = 0;
  52.             }
  53.         };
  54.  
  55.         DFS(0, -1);
  56.  
  57.         int ans = 0;
  58.  
  59.         for(int i=0; i<n; i++)
  60.         {
  61.             if(dis[i]!=0 && dis[i]<=ht[i])
  62.                 ans = max(ans, cnt[i]);
  63.         }
  64.  
  65.         cout<<ans<<"\n";
  66.     }
  67. }
Advertisement
Add Comment
Please, Sign In to add comment