oj.uz

Submission #1045557

# Submission time Handle Problem Language Result Execution time Memory
1045557 2024-08-06T05:39:41 Z HunterXD Mergers (JOI19_mergers) C++17
0 / 100
 44 ms 41936 KB 
#include <bits/stdc++.h>

using namespace std;

#define pb push_back

typedef pair<int, int> pi;

const char nd = '\n';

void solve();

signed main(){
  ios_base::sync_with_stdio(0);
  cin.tie(0);

  int t = 1;
  // cin >> t;
  while(t-->0) solve();
}

// ------------------- Code Starts here -------------------

// Union Find
struct UF{
  vector<int> p, s;
  UF(int n): p(n), s(n, 1){
    iota(p.begin(), p.end(), 0);
  }

  int find(int x){
    return p[x] == x ? x : p[x] = find(p[x]);
  }

  void join(int x, int y){
    x = find(x);
    y = find(y);
    if(x == y) return;
    if(s[x] < s[y]) swap(x, y);
    p[y] = x;
    s[x] += s[y];
  }
};


const int N = 5e5+7;

vector<int> g[N], state[N];
int s[N], parent[N], level[N], jump[N];
UF state_group(N);

// dfs to get parent and level of each node
void dfs(int u, int p){
  parent[u] = jump[u] = p;
  level[u] = level[p]+1;
  for (int v : g[u]){
    if (v == p) continue;
    dfs(v, u);
  }
}

void solve(){
  int n, k;
  cin >> n >> k;

  vector<pi> edges;

  for (int i = 1; i < n; i++){
    int u, v;
    cin >> u >> v;
    edges.pb({u, v});
    g[u].pb(v);
    g[v].pb(u);    
  }

  for (int i = 1; i <= n; i++){
    cin >> s[i];
    state[s[i]].pb(i);
  }

  dfs(1, 0);

  for (int i = 1; i <= n; i++){
    int mn = N;
    set<pi> st;

    for(int u: state[i]) st.insert({-level[u], u});
    
    while(st.size() > 1){
      int v;
      auto [l, u] = *st.begin();
      st.erase(st.begin());

      state_group.join(s[u], parent[u]);
      v = jump[u] = jump[parent[u]];
      st.insert({-level[v], v});
    }
  }

  // degree of each state group
  vector<int> deg(n+1, 0);

  for(auto [u, v]: edges){
    u = state_group.find(s[u]);
    v = state_group.find(s[v]);
    if (u == v) continue;
    deg[u]++;
    deg[v]++;
  }

  int leaf = 0;
  for(int i = 1; i <= n; i++){
    if(deg[i] == 1) leaf++;
  }

  cout << (leaf + 1) / 2 << nd;
}

Compilation message

mergers.cpp: In function 'void solve()':
mergers.cpp:84:9: warning: unused variable 'mn' [-Wunused-variable]
   84 |     int mn = N;
      |         ^~

Subtask #1
0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 5 ms 33372 KB Output is correct
2 Incorrect 5 ms 33372 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #2
0 / 24.0

# Verdict Execution time Memory Grader output
1 Correct 5 ms 33372 KB Output is correct
2 Incorrect 5 ms 33372 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #3
0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 5 ms 33372 KB Output is correct
2 Incorrect 5 ms 33372 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #4
0 / 22.0

# Verdict Execution time Memory Grader output
1 Correct 34 ms 39368 KB Output is correct
2 Correct 44 ms 41936 KB Output is correct
3 Incorrect 6 ms 33628 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #5
0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 5 ms 33372 KB Output is correct
2 Incorrect 5 ms 33372 KB Output isn't correct
3 Halted 0 ms 0 KB -