Advertisement
esraa_syam

Untitled

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