oj.uz

Submission #1030705

# Submission time Handle Problem Language Result Execution time Memory
1030705 2024-07-22T08:51:49 Z NeroZein Tourism (JOI23_tourism) C++17
10 / 100
 5000 ms 329252 KB 
#include "bits/stdc++.h"
using namespace std;

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

const int B = 400;
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 cnt[B][N];
int in_id[N * 2];
long long ans[N];
long long res[B];
int pre[B][N], nxt[B][N];
vector<pair<int, int>> qs[N]; 

void preprocess() {//B * N
  for (int ver = 0; ver * B < m; ++ver) {//M / B (B)
    for (int r = ver * B; r < m; ++r) {//M
      int v = c[r];
      cnt[ver][v]++;
      in_id[id[v]] = v; 
    }
    vector<int> ord; 
    for (int i = 0; i < idd; ++i) {//2 * N
      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) {//N
      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) {//M / B
    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;
  assert(l - cl < 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 2 ms 15704 KB Output is correct
2 Correct 2 ms 15452 KB Output is correct
3 Correct 2 ms 15452 KB Output is correct
4 Correct 5 ms 19548 KB Output is correct
5 Correct 5 ms 19684 KB Output is correct
6 Correct 4 ms 19692 KB Output is correct
7 Correct 6 ms 19548 KB Output is correct
8 Correct 4 ms 19548 KB Output is correct
9 Correct 9 ms 19844 KB Output is correct
10 Correct 10 ms 19804 KB Output is correct
11 Correct 8 ms 19804 KB Output is correct
12 Correct 3 ms 19548 KB Output is correct
13 Correct 4 ms 19548 KB Output is correct
14 Correct 4 ms 19548 KB Output is correct
15 Correct 8 ms 19800 KB Output is correct
16 Correct 8 ms 19848 KB Output is correct
17 Correct 7 ms 19804 KB Output is correct
18 Correct 11 ms 19804 KB Output is correct
19 Correct 8 ms 19804 KB Output is correct
20 Correct 9 ms 19804 KB Output is correct
21 Correct 7 ms 19804 KB Output is correct
22 Correct 7 ms 19804 KB Output is correct
23 Correct 8 ms 19804 KB Output is correct
24 Correct 8 ms 19624 KB Output is correct
25 Correct 8 ms 19800 KB Output is correct
26 Correct 7 ms 19804 KB Output is correct
27 Correct 2 ms 13416 KB Output is correct
28 Correct 2 ms 19548 KB Output is correct
29 Correct 8 ms 19800 KB Output is correct

Subtask #2
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 15704 KB Output is correct
2 Correct 2 ms 15452 KB Output is correct
3 Correct 2 ms 15452 KB Output is correct
4 Correct 5 ms 19548 KB Output is correct
5 Correct 5 ms 19684 KB Output is correct
6 Correct 4 ms 19692 KB Output is correct
7 Correct 6 ms 19548 KB Output is correct
8 Correct 4 ms 19548 KB Output is correct
9 Correct 9 ms 19844 KB Output is correct
10 Correct 10 ms 19804 KB Output is correct
11 Correct 8 ms 19804 KB Output is correct
12 Correct 3 ms 19548 KB Output is correct
13 Correct 4 ms 19548 KB Output is correct
14 Correct 4 ms 19548 KB Output is correct
15 Correct 8 ms 19800 KB Output is correct
16 Correct 8 ms 19848 KB Output is correct
17 Correct 7 ms 19804 KB Output is correct
18 Correct 11 ms 19804 KB Output is correct
19 Correct 8 ms 19804 KB Output is correct
20 Correct 9 ms 19804 KB Output is correct
21 Correct 7 ms 19804 KB Output is correct
22 Correct 7 ms 19804 KB Output is correct
23 Correct 8 ms 19804 KB Output is correct
24 Correct 8 ms 19624 KB Output is correct
25 Correct 8 ms 19800 KB Output is correct
26 Correct 7 ms 19804 KB Output is correct
27 Correct 2 ms 13416 KB Output is correct
28 Correct 2 ms 19548 KB Output is correct
29 Correct 8 ms 19800 KB Output is correct
30 Correct 236 ms 29828 KB Output is correct
31 Correct 390 ms 29888 KB Output is correct
32 Correct 607 ms 32200 KB Output is correct
33 Correct 575 ms 32456 KB Output is correct
34 Correct 695 ms 33220 KB Output is correct
35 Correct 77 ms 28536 KB Output is correct
36 Correct 81 ms 28368 KB Output is correct
37 Correct 75 ms 28452 KB Output is correct
38 Correct 598 ms 32700 KB Output is correct
39 Correct 559 ms 32704 KB Output is correct
40 Correct 568 ms 33784 KB Output is correct
41 Correct 78 ms 28620 KB Output is correct
42 Correct 85 ms 28716 KB Output is correct
43 Correct 86 ms 28648 KB Output is correct
44 Correct 588 ms 32708 KB Output is correct
45 Correct 576 ms 33472 KB Output is correct
46 Correct 584 ms 33968 KB Output is correct
47 Correct 67 ms 28252 KB Output is correct
48 Correct 84 ms 28364 KB Output is correct
49 Correct 89 ms 28368 KB Output is correct
50 Correct 631 ms 32708 KB Output is correct
51 Correct 613 ms 33524 KB Output is correct
52 Correct 628 ms 32708 KB Output is correct
53 Correct 650 ms 33732 KB Output is correct
54 Correct 685 ms 32704 KB Output is correct
55 Correct 595 ms 33108 KB Output is correct
56 Correct 7 ms 19548 KB Output is correct
57 Correct 3 ms 21596 KB Output is correct
58 Correct 433 ms 30076 KB Output is correct

Subtask #3
0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 15452 KB Output is correct
2 Correct 2 ms 13404 KB Output is correct
3 Correct 7 ms 19544 KB Output is correct
4 Execution timed out 5097 ms 197100 KB Time limit exceeded
5 Halted 0 ms 0 KB -

Subtask #4
0 / 18.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 15448 KB Output is correct
2 Correct 971 ms 125824 KB Output is correct
3 Correct 1482 ms 195712 KB Output is correct
4 Correct 1164 ms 170580 KB Output is correct
5 Correct 931 ms 324596 KB Output is correct
6 Correct 893 ms 324644 KB Output is correct
7 Correct 916 ms 324844 KB Output is correct
8 Correct 1008 ms 324668 KB Output is correct
9 Correct 935 ms 324904 KB Output is correct
10 Correct 983 ms 325268 KB Output is correct
11 Correct 1278 ms 326616 KB Output is correct
12 Correct 3148 ms 328796 KB Output is correct
13 Execution timed out 5064 ms 329252 KB Time limit exceeded
14 Halted 0 ms 0 KB -

Subtask #5
0 / 24.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 15428 KB Output is correct
2 Correct 2 ms 13400 KB Output is correct
3 Correct 7 ms 19548 KB Output is correct
4 Execution timed out 5105 ms 195776 KB Time limit exceeded
5 Halted 0 ms 0 KB -

Subtask #6
0 / 41.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 15704 KB Output is correct
2 Correct 2 ms 15452 KB Output is correct
3 Correct 2 ms 15452 KB Output is correct
4 Correct 5 ms 19548 KB Output is correct
5 Correct 5 ms 19684 KB Output is correct
6 Correct 4 ms 19692 KB Output is correct
7 Correct 6 ms 19548 KB Output is correct
8 Correct 4 ms 19548 KB Output is correct
9 Correct 9 ms 19844 KB Output is correct
10 Correct 10 ms 19804 KB Output is correct
11 Correct 8 ms 19804 KB Output is correct
12 Correct 3 ms 19548 KB Output is correct
13 Correct 4 ms 19548 KB Output is correct
14 Correct 4 ms 19548 KB Output is correct
15 Correct 8 ms 19800 KB Output is correct
16 Correct 8 ms 19848 KB Output is correct
17 Correct 7 ms 19804 KB Output is correct
18 Correct 11 ms 19804 KB Output is correct
19 Correct 8 ms 19804 KB Output is correct
20 Correct 9 ms 19804 KB Output is correct
21 Correct 7 ms 19804 KB Output is correct
22 Correct 7 ms 19804 KB Output is correct
23 Correct 8 ms 19804 KB Output is correct
24 Correct 8 ms 19624 KB Output is correct
25 Correct 8 ms 19800 KB Output is correct
26 Correct 7 ms 19804 KB Output is correct
27 Correct 2 ms 13416 KB Output is correct
28 Correct 2 ms 19548 KB Output is correct
29 Correct 8 ms 19800 KB Output is correct
30 Correct 236 ms 29828 KB Output is correct
31 Correct 390 ms 29888 KB Output is correct
32 Correct 607 ms 32200 KB Output is correct
33 Correct 575 ms 32456 KB Output is correct
34 Correct 695 ms 33220 KB Output is correct
35 Correct 77 ms 28536 KB Output is correct
36 Correct 81 ms 28368 KB Output is correct
37 Correct 75 ms 28452 KB Output is correct
38 Correct 598 ms 32700 KB Output is correct
39 Correct 559 ms 32704 KB Output is correct
40 Correct 568 ms 33784 KB Output is correct
41 Correct 78 ms 28620 KB Output is correct
42 Correct 85 ms 28716 KB Output is correct
43 Correct 86 ms 28648 KB Output is correct
44 Correct 588 ms 32708 KB Output is correct
45 Correct 576 ms 33472 KB Output is correct
46 Correct 584 ms 33968 KB Output is correct
47 Correct 67 ms 28252 KB Output is correct
48 Correct 84 ms 28364 KB Output is correct
49 Correct 89 ms 28368 KB Output is correct
50 Correct 631 ms 32708 KB Output is correct
51 Correct 613 ms 33524 KB Output is correct
52 Correct 628 ms 32708 KB Output is correct
53 Correct 650 ms 33732 KB Output is correct
54 Correct 685 ms 32704 KB Output is correct
55 Correct 595 ms 33108 KB Output is correct
56 Correct 7 ms 19548 KB Output is correct
57 Correct 3 ms 21596 KB Output is correct
58 Correct 433 ms 30076 KB Output is correct
59 Correct 2 ms 15452 KB Output is correct
60 Correct 2 ms 13404 KB Output is correct
61 Correct 7 ms 19544 KB Output is correct
62 Execution timed out 5097 ms 197100 KB Time limit exceeded
63 Halted 0 ms 0 KB -