oj.uz

Submission #1030529

# Submission time Handle Problem Language Result Execution time Memory
1030529 2024-07-22T06:29:30 Z NeroZein Tourism (JOI23_tourism) C++17
10 / 100
 543 ms 182884 KB 
#include "bits/stdc++.h"
using namespace std;

#ifdef Nero
#include "Deb.h"
#else
#define debug(...)
#endif

const int B = 320;
const int LOG = 18; 
const int N = 1e5 + 5; 

int n, m, q;
int c[N];
int dep[N];
int idd, id[N];
vector<int> g[N];
int spa[LOG][N * 2]; 

void dfs(int v, int p) {
  id[v] = idd;
  spa[0][idd++] = v;
  for (int u : g[v]) {
    if (u == p) {
      continue; 
    }
    dep[u] = dep[v] + 1; 
    dfs(u, v);
    spa[0][idd++] = v; 
  }
}

int get(int l, int r) {
  int lg = 31 - __builtin_clz(r - l + 1);
  int u = spa[lg][l], v = spa[lg][r - (1 << lg) + 1];
  return dep[u] < dep[v] ? u : v;
}

int distance(int u, int v) {
  int l = id[u], r = id[v];
  if (l > r) {
    swap(l, r); 
  }
  int lca = get(l, r);
  int ret = dep[u] + dep[v] - 2 * dep[lca];
  return ret; 
}

int in_id[N];
int cnt[100][N];
long long ans[N];
long long res[100];
int pre[100][N], nxt[100][N];
vector<pair<int, int>> qs[N]; 

void preprocess() {
  for (int ver = 0; ver * B < m; ++ver) {
    for (int r = ver * B; r < m; ++r) {
      int v = c[r];
      cnt[ver][v]++;
      in_id[id[v]] = v; 
    }
    vector<int> ord; 
    for (int i = 0; i < idd; ++i) {
      if (in_id[i] != 0) {
        ord.push_back(in_id[i]);
        in_id[i] = 0; 
      }
    }
    //debug(ord);
    //cerr << '\n';
    for (int i = 1; i < (int) ord.size(); ++i) {
      int u = ord[i - 1], v = ord[i];
      res[ver] += distance(u, v); 
      pre[ver][v] = u;
      nxt[ver][u] = v; 
    }
    nxt[ver][ord.back()] = ord[0];
    pre[ver][ord[0]] = ord.back();
    res[ver] += distance(ord[0], ord.back()); 
  }
}

void back(int r) {
  //debug(r);
  for (int ver = 0; ver * B <= r; ++ver) {
    int v = c[r];
    if (--cnt[ver][v] > 0) {
      continue; 
    }
    //int old = res[ver];
    int cpre = pre[ver][v];
    int cnxt = nxt[ver][v];
    if (cpre != 0) {
      res[ver] -= distance(v, cpre);
    }
    if (cnxt != 0) {
      res[ver] -= distance(v, cnxt);
    }
    if (cpre != 0 && cnxt != 0) {
      nxt[ver][cpre] = cnxt;
      pre[ver][cnxt] = cpre; 
      res[ver] += distance(cpre, cnxt); 
    }
    //debug(ver, v, cpre, cnxt, old, res[ver]);
  }
  //cerr << '\n';
}

int cnt2[N];
vector<int> clean;
int curpre[N], curnxt[N];

void check(int v) {
  if (curpre[v] == 0 && curnxt[v] == 0) {
    clean.push_back(v);
  }
}

void rem(int ver, int cl, long long& ret) {
  int v = c[cl];
  check(v);
  if (cnt2[v] == 0) {
    cnt2[v] = cnt[ver][v];
  }
  if (curpre[v] == 0) {
    curpre[v] = pre[ver][v];
  }
  if (curnxt[v] == 0) {
    curnxt[v] = nxt[ver][v];
  }
  if (--cnt2[v] != 0) {
    return; 
  }
  int cpre = curpre[v];
  int cnxt = curnxt[v];
  if (cpre != 0) {
    ret -= distance(v, cpre);
  }
  if (cnxt != 0) {
    ret -= distance(v, cnxt);
  }
  if (cpre != 0 && cnxt != 0) {
    check(cpre);
    check(cnxt);
    curpre[cnxt] = cpre;
    curnxt[cpre] = cnxt; 
    ret += distance(cpre, cnxt); 
  }
}

long long qry(int l) {
  int ver = l / B;
  long long ret = res[ver];
  int cl = ver * B;
  while (cl < l) {
    rem(ver, cl, ret);
    cl++;
  }
  return ret; 
}

void cleanup() {
  for (int v : clean) {
    cnt2[v] = curpre[v] = curnxt[v] = 0; 
  }
}

int main() {
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  cin >> n >> m >> q;
  for(int i = 0; i < n - 1; ++i) {
    int u, v;
    cin >> u >> v;
    g[u].push_back(v);
    g[v].push_back(u);
  }
  dfs(1, 1); 
  for (int j = 1; j < LOG; ++j) {
    for (int i = 0; i + (1 << j) <= idd; ++i) {
      int u = spa[j - 1][i], v = spa[j - 1][i + (1 << (j - 1))];
      spa[j][i] = (dep[u] < dep[v] ? u : v); 
    }
  }
  for (int i = 0; i < m; ++i) {
    cin >> c[i];
  }
  preprocess();
  for (int i = 0; i < q; ++i) {
    int l, r;
    cin >> l >> r;
    --l, --r;
    qs[r].push_back({l, i});
  }
  for (int r = m - 1; r >= 0; --r) {
    for (auto [ql, qi] : qs[r]) {
      ans[qi] = (qry(ql) + 2) / 2;
      cleanup();
    }
    back(r); 
  }
  for (int i = 0; i < q; ++i) {
    cout << ans[i] << '\n';
  }
  return 0;
}

Subtask #1
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 5212 KB Output is correct
2 Correct 2 ms 5212 KB Output is correct
3 Correct 3 ms 5212 KB Output is correct
4 Correct 7 ms 5212 KB Output is correct
5 Correct 7 ms 5212 KB Output is correct
6 Correct 4 ms 5212 KB Output is correct
7 Correct 6 ms 5468 KB Output is correct
8 Correct 4 ms 5424 KB Output is correct
9 Correct 12 ms 5468 KB Output is correct
10 Correct 9 ms 5468 KB Output is correct
11 Correct 9 ms 5468 KB Output is correct
12 Correct 3 ms 5212 KB Output is correct
13 Correct 3 ms 5212 KB Output is correct
14 Correct 3 ms 5212 KB Output is correct
15 Correct 9 ms 5468 KB Output is correct
16 Correct 10 ms 5552 KB Output is correct
17 Correct 9 ms 5472 KB Output is correct
18 Correct 9 ms 5464 KB Output is correct
19 Correct 8 ms 5468 KB Output is correct
20 Correct 10 ms 5568 KB Output is correct
21 Correct 11 ms 5724 KB Output is correct
22 Correct 8 ms 5440 KB Output is correct
23 Correct 8 ms 5572 KB Output is correct
24 Correct 11 ms 5436 KB Output is correct
25 Correct 11 ms 5464 KB Output is correct
26 Correct 14 ms 5560 KB Output is correct
27 Correct 2 ms 5212 KB Output is correct
28 Correct 2 ms 5212 KB Output is correct
29 Correct 10 ms 5464 KB Output is correct

Subtask #2
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 5212 KB Output is correct
2 Correct 2 ms 5212 KB Output is correct
3 Correct 3 ms 5212 KB Output is correct
4 Correct 7 ms 5212 KB Output is correct
5 Correct 7 ms 5212 KB Output is correct
6 Correct 4 ms 5212 KB Output is correct
7 Correct 6 ms 5468 KB Output is correct
8 Correct 4 ms 5424 KB Output is correct
9 Correct 12 ms 5468 KB Output is correct
10 Correct 9 ms 5468 KB Output is correct
11 Correct 9 ms 5468 KB Output is correct
12 Correct 3 ms 5212 KB Output is correct
13 Correct 3 ms 5212 KB Output is correct
14 Correct 3 ms 5212 KB Output is correct
15 Correct 9 ms 5468 KB Output is correct
16 Correct 10 ms 5552 KB Output is correct
17 Correct 9 ms 5472 KB Output is correct
18 Correct 9 ms 5464 KB Output is correct
19 Correct 8 ms 5468 KB Output is correct
20 Correct 10 ms 5568 KB Output is correct
21 Correct 11 ms 5724 KB Output is correct
22 Correct 8 ms 5440 KB Output is correct
23 Correct 8 ms 5572 KB Output is correct
24 Correct 11 ms 5436 KB Output is correct
25 Correct 11 ms 5464 KB Output is correct
26 Correct 14 ms 5560 KB Output is correct
27 Correct 2 ms 5212 KB Output is correct
28 Correct 2 ms 5212 KB Output is correct
29 Correct 10 ms 5464 KB Output is correct
30 Correct 205 ms 7672 KB Output is correct
31 Correct 364 ms 7840 KB Output is correct
32 Correct 472 ms 7888 KB Output is correct
33 Correct 479 ms 8140 KB Output is correct
34 Correct 489 ms 8144 KB Output is correct
35 Correct 75 ms 6504 KB Output is correct
36 Correct 83 ms 6348 KB Output is correct
37 Correct 75 ms 6352 KB Output is correct
38 Correct 488 ms 8124 KB Output is correct
39 Correct 520 ms 8136 KB Output is correct
40 Correct 492 ms 8136 KB Output is correct
41 Correct 75 ms 6640 KB Output is correct
42 Correct 79 ms 6608 KB Output is correct
43 Correct 92 ms 6544 KB Output is correct
44 Correct 476 ms 8260 KB Output is correct
45 Correct 495 ms 10180 KB Output is correct
46 Correct 491 ms 8236 KB Output is correct
47 Correct 69 ms 6236 KB Output is correct
48 Correct 77 ms 6348 KB Output is correct
49 Correct 81 ms 6604 KB Output is correct
50 Correct 539 ms 8180 KB Output is correct
51 Correct 530 ms 7880 KB Output is correct
52 Correct 499 ms 7884 KB Output is correct
53 Correct 543 ms 7904 KB Output is correct
54 Correct 536 ms 7876 KB Output is correct
55 Correct 515 ms 8116 KB Output is correct
56 Correct 7 ms 5208 KB Output is correct
57 Correct 3 ms 5468 KB Output is correct
58 Correct 329 ms 7972 KB Output is correct

Subtask #3
0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 5212 KB Output is correct
2 Correct 3 ms 5160 KB Output is correct
3 Correct 6 ms 5212 KB Output is correct
4 Runtime error 231 ms 182884 KB Execution killed with signal 11
5 Halted 0 ms 0 KB -

Subtask #4
0 / 18.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 5212 KB Output is correct
2 Runtime error 211 ms 172316 KB Execution killed with signal 11
3 Halted 0 ms 0 KB -

Subtask #5
0 / 24.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 5212 KB Output is correct
2 Correct 3 ms 5212 KB Output is correct
3 Correct 6 ms 5244 KB Output is correct
4 Runtime error 239 ms 176116 KB Execution killed with signal 11
5 Halted 0 ms 0 KB -

Subtask #6
0 / 41.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 5212 KB Output is correct
2 Correct 2 ms 5212 KB Output is correct
3 Correct 3 ms 5212 KB Output is correct
4 Correct 7 ms 5212 KB Output is correct
5 Correct 7 ms 5212 KB Output is correct
6 Correct 4 ms 5212 KB Output is correct
7 Correct 6 ms 5468 KB Output is correct
8 Correct 4 ms 5424 KB Output is correct
9 Correct 12 ms 5468 KB Output is correct
10 Correct 9 ms 5468 KB Output is correct
11 Correct 9 ms 5468 KB Output is correct
12 Correct 3 ms 5212 KB Output is correct
13 Correct 3 ms 5212 KB Output is correct
14 Correct 3 ms 5212 KB Output is correct
15 Correct 9 ms 5468 KB Output is correct
16 Correct 10 ms 5552 KB Output is correct
17 Correct 9 ms 5472 KB Output is correct
18 Correct 9 ms 5464 KB Output is correct
19 Correct 8 ms 5468 KB Output is correct
20 Correct 10 ms 5568 KB Output is correct
21 Correct 11 ms 5724 KB Output is correct
22 Correct 8 ms 5440 KB Output is correct
23 Correct 8 ms 5572 KB Output is correct
24 Correct 11 ms 5436 KB Output is correct
25 Correct 11 ms 5464 KB Output is correct
26 Correct 14 ms 5560 KB Output is correct
27 Correct 2 ms 5212 KB Output is correct
28 Correct 2 ms 5212 KB Output is correct
29 Correct 10 ms 5464 KB Output is correct
30 Correct 205 ms 7672 KB Output is correct
31 Correct 364 ms 7840 KB Output is correct
32 Correct 472 ms 7888 KB Output is correct
33 Correct 479 ms 8140 KB Output is correct
34 Correct 489 ms 8144 KB Output is correct
35 Correct 75 ms 6504 KB Output is correct
36 Correct 83 ms 6348 KB Output is correct
37 Correct 75 ms 6352 KB Output is correct
38 Correct 488 ms 8124 KB Output is correct
39 Correct 520 ms 8136 KB Output is correct
40 Correct 492 ms 8136 KB Output is correct
41 Correct 75 ms 6640 KB Output is correct
42 Correct 79 ms 6608 KB Output is correct
43 Correct 92 ms 6544 KB Output is correct
44 Correct 476 ms 8260 KB Output is correct
45 Correct 495 ms 10180 KB Output is correct
46 Correct 491 ms 8236 KB Output is correct
47 Correct 69 ms 6236 KB Output is correct
48 Correct 77 ms 6348 KB Output is correct
49 Correct 81 ms 6604 KB Output is correct
50 Correct 539 ms 8180 KB Output is correct
51 Correct 530 ms 7880 KB Output is correct
52 Correct 499 ms 7884 KB Output is correct
53 Correct 543 ms 7904 KB Output is correct
54 Correct 536 ms 7876 KB Output is correct
55 Correct 515 ms 8116 KB Output is correct
56 Correct 7 ms 5208 KB Output is correct
57 Correct 3 ms 5468 KB Output is correct
58 Correct 329 ms 7972 KB Output is correct
59 Correct 2 ms 5212 KB Output is correct
60 Correct 3 ms 5160 KB Output is correct
61 Correct 6 ms 5212 KB Output is correct
62 Runtime error 231 ms 182884 KB Execution killed with signal 11
63 Halted 0 ms 0 KB -