Advertisement
El_GEMMY

tree xor

Apr 28th, 2022
63
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 4.50 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 3.14159265
  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 finmod(x, m) x = ((x) % (m) + (m)) % m
  38. #define debug(x) cout << "x: " << (x) << nl;
  39. #define debug2(x, y) cout << "x: " << (x) << " y: " << y << nl;
  40. #define ordered_set tree<pair<int, int>, null_type, less<>, rb_tree_tag, tree_order_statistics_node_update>
  41. #define ordered_map tree<int, int, less<>, rb_tree_tag, tree_order_statistics_node_update>
  42.  
  43. //vector<int> dx = {0, 0, 1, -1, 1, 1, -1, -1}, dy = {1, -1, 0, 0, 1, -1, 1, -1};
  44. //vector<int> dx = {0, 0, 1, -1}, dy = {1, -1, 0, 0};
  45.  
  46. template<typename T = int> istream& operator>>(istream& in, vector<pair<int, int>>& v){
  47.     for (auto& [x, y] : v) in >> x >> y;
  48.     return in;
  49. }
  50.  
  51. template<typename T = int> istream& operator>>(istream& in, vector<T>& v){
  52.     for (T& i : v) in >> i;
  53.     return in;
  54. }
  55.  
  56. template<typename T = int> ostream& operator<<(ostream& out, const vector<T>& v){
  57.     for (const T& x : v)
  58.         out << x << ' ';
  59.     return out;
  60. }
  61.  
  62. template<typename T = pair<int, int>> ostream& operator << (ostream& out, const vector<pair<int, int>>& v){
  63.     for(auto& [x, y] : v){
  64.         out << x << ' ' << y << nl;
  65.     }
  66.     return out;
  67. }
  68.  
  69. void Start_Crushing() {
  70.     ios_base::sync_with_stdio(false);
  71.     cin.tie(nullptr);
  72.     cout.tie(nullptr);
  73. #ifndef ONLINE_JUDGE
  74.     freopen("input.txt", "r", stdin);
  75.     freopen("output.txt", "w", stdout);
  76. #endif
  77. }
  78. vector<vector<pair<int, int>>> tree_1, tree_2;
  79. vector<int> one_1, one_2, zero_1, zero_2;
  80. vector<bool> vis;
  81.  
  82. void dfs(int src, int st, int curr, bool one, vector<vector<pair<int, int>>>& tree){
  83.  
  84.     for(int bit = 0; bit < 31; bit++){
  85.         if(curr & (1 << bit)){
  86.             (one ? one_1 : one_2)[bit]++;
  87.         }else{
  88.             (one ? zero_1 : zero_2)[bit]++;
  89.         }
  90.     }
  91.  
  92.     vis[src] = true;
  93.  
  94.     for(auto& [nxt, wgt] : tree[src]){
  95.         if(not vis[nxt]){
  96.             dfs(nxt, st, curr ^ wgt, one, tree);
  97.         }
  98.     }
  99. }
  100.  
  101. void solve(){
  102.     int n; cin >> n;
  103.     tree_1.assign(n + 5, vector<pair<int, int>>());
  104.     tree_2.assign(n + 5, vector<pair<int, int>>());
  105.  
  106.     one_1.resize(32), one_2.resize(32), zero_1.resize(32), zero_2.resize(32);
  107.  
  108.     for(int i = 0; i < 2; i++){
  109.         for(int j = 0; j < n - 1; j++){
  110.             int u, v, w; cin >> u >> v >> w;
  111.             (i & 1 ? tree_2 : tree_1)[u].emplace_back(v, w);
  112.             (i & 1 ? tree_2 : tree_1)[v].emplace_back(u, w);
  113.         }
  114.     }
  115.  
  116.     // we should count the number of ways in which we can set this bit to one in both trees
  117.     // and then, I think we can multiply the two frequencies by each others
  118.     for(int i = 1; i <= n; i++){
  119.         vis.assign(n + 5, false);
  120.         dfs(i, i, 0, true, tree_1);
  121.         vis.assign(n + 5, false);
  122.         dfs(i, i, 0, false, tree_2);
  123.     }
  124.  
  125.     int sum = 0;
  126.     debug(one_1)
  127.     debug(zero_1)
  128.     debug(one_2)
  129.     debug(zero_2)
  130.  
  131.     for(int i = 1; i <= n; i++){
  132.         for(int j = 1; j <= n; j++){
  133.             for(int k = 1; k <= n; k++){
  134.                 if(i == j and j == k)
  135.                     continue;
  136.                 cout << i << ' ' << j << ' ' << k << nl;
  137.             }
  138.         }
  139.     }
  140.  
  141.    
  142. }
  143.  
  144. int main(){
  145.     Start_Crushing();
  146.  
  147.     int t = 1;
  148. //    /*Multiple test cases?*/ cin >> t;
  149.     while (t--) {
  150.         solve();
  151.         if(!t)
  152.             break;
  153.         cout << nl;
  154.     }
  155.  
  156. //    for(int tc = 1; tc <= t; tc++){
  157. //        cout << "Case #" << tc << ": ";
  158. //        solve();
  159. //        if(tc != t)
  160. //            cout << nl;
  161. //    }
  162.  
  163.     return 0;
  164. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement