Advertisement
El_GEMMY

Tree Diameter #5

Jan 23rd, 2023
814
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 4.44 KB | None | 0 0
  1. // Those who cannot remember the past are
  2. // condemned to repeat it (use DP -_-)
  3. // - George Santayana
  4.  
  5. #include <bits/stdc++.h>
  6. #include <ext/pb_ds/assoc_container.hpp>
  7. #include <ext/pb_ds/tree_policy.hpp>
  8.  
  9. using namespace std;
  10. using namespace __gnu_pbds;
  11.  
  12. #define all(v) v.begin(), v.end()
  13. #define rall(v) v.rbegin(), v.rend()
  14. #define ll long long
  15. #define ull unsigned long long
  16. #define MOD 1000000007
  17. #define PI acos(-1)
  18. #define ceil(a, b) (((a) / (b)) + ((a) % (b) ? 1 : 0))
  19. #define imin INT_MIN
  20. #define imax INT_MAX
  21. #define llmax LLONG_MAX
  22. #define llmin LLONG_MIN
  23. #define inf 2000000000
  24. #define nl '\n'
  25. #define ppcnt __builtin_popcount
  26. #define ppcntll __builtin_popcountll
  27. #define clz __builtin_clz
  28. #define clzll __builtin_clzll
  29. #define ctz __builtin_ctz
  30. #define ctzll __builtin_ctzll
  31. #define modulo(a, b, mod) ((((a) % (mod)) + ((b) % (mod))) % (mod))
  32. #define cnte(v, x) count(all(v), (x))
  33. #define mine(v) min_element(all(v))
  34. #define maxe(v) max_element(all(v))
  35. #define updmin(a, b) a = min(a, b)
  36. #define updmax(a, b) a = max(a, b)
  37. #define findmod(x, m) x = ((x) % (m) + (m)) % m
  38. #define getmod(x, m) ((x) % (m) + (m)) % (m)
  39. #define debug(x) cout << "x: " << (x) << nl;
  40. #define debug2(x, y) cout << "x: " << (x) << " y: " << y << nl;
  41. #define ordered_set tree<int, null_type, less<>, rb_tree_tag, tree_order_statistics_node_update>
  42. #define ordered_map tree<int, int, less<>, rb_tree_tag, tree_order_statistics_node_update>
  43.  
  44. //vector<int> dx = {0, 0, 1, -1, 1, 1, -1, -1}, dy = {1, -1, 0, 0, 1, -1, 1, -1};
  45. //vector<int> dx = {0, 0, 1, -1}, dy = {1, -1, 0, 0};
  46.  
  47. template<typename T = int> istream& operator>>(istream& in, vector<pair<int, int>>& v){
  48.     for (auto& [x, y] : v) in >> x >> y;
  49.     return in;
  50. }
  51.  
  52. template<typename T = int> istream& operator>>(istream& in, vector<T>& v){
  53.     for (T& i : v) in >> i;
  54.     return in;
  55. }
  56.  
  57. template<typename T = int> ostream& operator<<(ostream& out, const vector<T>& v){
  58.     for (const T& x : v)
  59.         out << x << ' ';
  60.     return out;
  61. }
  62.  
  63. template<typename T = pair<int, int>> ostream& operator << (ostream& out, const vector<pair<int, int>>& v){
  64.     for(auto& [x, y] : v){
  65.         out << x << ' ' << y << nl;
  66.     }
  67.     return out;
  68. }
  69.  
  70. void Start_Crushing() {
  71.     ios_base::sync_with_stdio(false);
  72.     cin.tie(nullptr);
  73.     cout.tie(nullptr);
  74. #ifndef ONLINE_JUDGE
  75.     freopen("input.txt", "r", stdin);
  76.     freopen("output.txt", "w", stdout);
  77. #endif
  78. }
  79. int n;
  80. vector<vector<int>> adj, memo;
  81.  
  82. int dfs(int src, int par){
  83.     int& ret = memo[src][par];
  84.     if(~ret)
  85.         return ret;
  86.  
  87.     ret = 0;
  88.     for(auto& nxt : adj[src]){
  89.         if(nxt == par)
  90.             continue;
  91.         updmax(ret, dfs(nxt, src));
  92.     }
  93.     return ++ret;
  94. }
  95.  
  96. void solve(){
  97.     while(cin >> n){
  98.         adj.assign(n + 5, vector<int>());
  99.         memo.assign(n + 5, vector<int>(n + 5, -1));
  100.  
  101.         for(int i = 1; i <= n; i++){
  102.             int k; cin >> k;
  103.             while(k--){
  104.                 int v; cin >> v;
  105.                 adj[i].emplace_back(v);
  106.             }
  107.         }
  108.  
  109.         vector<int> worst, best;
  110.  
  111.         int mini = imax, maxi = imin;
  112.         for(int i = 1; i <= n; i++){
  113.             int curr = dfs(i, 0);
  114.             updmin(mini, curr);
  115.             updmax(maxi, curr);
  116.         }
  117.  
  118.         for(int i = 1; i <= n; i++){
  119.             int curr = *maxe(memo[i]);
  120.             if(curr == mini)
  121.                 best.emplace_back(i);
  122.             else if(curr == maxi)
  123.                 worst.emplace_back(i);
  124.         }
  125.  
  126.         cout << "Best Roots  : ";
  127.         for(int i = 0; i < best.size(); i++){
  128.             if(i)
  129.                 cout << ' ';
  130.             cout << best[i];
  131.         }
  132.  
  133.         cout << nl << "Worst Roots : ";
  134.         for(int i = 0; i < worst.size(); i++){
  135.             if(i)
  136.                 cout << ' ';
  137.             cout << worst[i];
  138.         }
  139.         cout << nl;
  140.     }
  141. }
  142.  
  143. void set_file(string& file_name){
  144.     freopen((file_name + ".in").c_str(), "r", stdin);
  145.     freopen((file_name + ".out").c_str(), "w", stdout);
  146. }
  147.  
  148. int main(){
  149.     Start_Crushing();
  150.  
  151. //    string file_name = "family";
  152. //    set_file(file_name);
  153.  
  154.     int t = 1;
  155. //    /*Multiple test cases?*/ cin >> t;
  156.     while (t--) {
  157.         solve();
  158.         if(!t)
  159.             break;
  160.         cout << nl;
  161.     }
  162.  
  163. //    for(int tc = 1; tc <= t; tc++){
  164. //        cout << "Case #" << tc << ": ";
  165. //        solve();
  166. //        if(tc != t)
  167. //            cout << nl;
  168. //    }
  169.  
  170.     return 0;
  171. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement