oj.uz

Submission #995119

# Submission time Handle Problem Language Result Execution time Memory
995119 2024-06-08T13:21:49 Z Szil Jail (JOI22_jail) C++17
5 / 100
 405 ms 175696 KB 
#include <bits/stdc++.h>

using namespace std;
using ll = long long;

const int MAXN = 200'100;

vector<int> g[MAXN], g2[MAXN*21];
int pos[MAXN], tin[MAXN], tout[MAXN], head[MAXN], heavy[MAXN], sz[MAXN], par[MAXN], 
    s[MAXN], e[MAXN], depth[MAXN], col[MAXN*21], timer, pos_timer, n, m;
bool ok;

void init() {
    for (int i = 1; i <= n; i++) {
        g[i].clear();
    }
    for (int i = 1; i <= 20*n; i++) {
        g2[i].clear();
    }
    fill(heavy, heavy+n+1, 0);
    fill(s, s+n+1, 0);
    fill(e, e+n+1, 0);
    fill(col, col+20*n+1, 0);
    timer = pos_timer = 1;
    ok = true;
}

int get_segtree1_index(int u) {
    return 10*n + u;
}

int get_segtree2_index(int u) {
    return n + u;
}

int get_path_index(int u) {
    return u;
}

void dfs(int u, int p = -1, int d = 0) {
    tin[u] = timer++;
    sz[u] = 1;
    depth[u] = d;
    for (int v : g[u]) {
        if (v == p) continue;
        par[v] = u;
        dfs(v, u, d+1);
        sz[u] += sz[v];
        if (sz[heavy[u]] < sz[v]) {
            heavy[u] = v;
        }
    }
    tout[u] = timer++;
}

bool is_ancestor(int u, int v) {
    return tin[u] <= tin[v] && tout[v] <= tout[u];
}

int nxt(int u, int v) {
    if (is_ancestor(u, v)) {
        for (int x : g[u]) {
            if (x != par[u] && is_ancestor(x, v)) return x;
        }
    } else {
        return par[u];
    }
}

void dfs2(int u, int h, int p) {
    head[u] = h;
    pos[u] = pos_timer++;
    if (heavy[u]) {
        dfs2(heavy[u], h, u);
    }
    for (int v : g[u]) {
        if (v == p || v == heavy[u]) continue;
        dfs2(v, v, u);
    }
}

void dfs3(int u) {
    col[u] = 1;
    for (int v : g2[u]) {
        if (col[v] == 0) dfs3(v);
        else if (col[v] == 1) ok = false;
    }
    col[u] = 2;
}

void build_segtree1(int v, int tl, int tr) {
    if (tl == tr) {
        if (s[tl]) {
            g2[get_segtree1_index(v)].emplace_back(get_path_index(s[tl]));
        }
    } else {
        int tm = (tl + tr) / 2;
        build_segtree1(2*v, tl, tm);
        build_segtree1(2*v+1, tm+1, tr);
        g2[get_segtree1_index(v)].emplace_back(get_segtree1_index(2*v));
        g2[get_segtree1_index(v)].emplace_back(get_segtree1_index(2*v+1));
    }
}

void build_segtree2(int v, int tl, int tr) {
    if (tl == tr) {
        if (e[tl]) {
            g2[get_path_index(e[tl])].emplace_back(get_segtree2_index(v));
        }
    } else {
        int tm = (tl + tr) / 2;
        build_segtree2(2*v, tl, tm);
        build_segtree2(2*v+1, tm+1, tr);
        g2[get_segtree2_index(2*v)].emplace_back(get_segtree2_index(v));
        g2[get_segtree2_index(2*v+1)].emplace_back(get_segtree2_index(v));
    }
}

void add_segtree1_edge(int v, int tl, int tr, int l, int r, int node) {
    if (l > r) return;
    if (l == tl && r == tr) {
        g2[node].emplace_back(get_segtree1_index(v));
    } else {
        int tm = (tl + tr) / 2;
        add_segtree1_edge(2*v, tl, tm, l, min(tm, r), node);
        add_segtree1_edge(2*v+1, tm+1, tr, max(tm+1, l), r, node);
    }
}

void add_segtree2_edge(int v, int tl, int tr, int l, int r, int node) {
    if (l > r) return;
    if (l == tl && r == tr) {
        g2[get_segtree2_index(v)].emplace_back(node);
    } else {
        int tm = (tl + tr) / 2;
        add_segtree2_edge(2*v, tl, tm, l, min(tm, r), node);
        add_segtree2_edge(2*v+1, tm+1, tr, max(tm+1, l), r, node);
    }
}

void qry1(int u, int v, int node) {
    while (head[u] != head[v]) {
        if (depth[head[u]] < depth[head[v]]) swap(u, v);
        add_segtree1_edge(1, 1, n, pos[head[u]], pos[u], node);
        u = par[head[u]];
    }
    if (depth[u] > depth[v]) swap(u, v);
    add_segtree1_edge(1, 1, n, pos[u], pos[v], node);
}

void qry2(int u, int v, int node) {
    while (head[u] != head[v]) {
        if (depth[head[u]] < depth[head[v]]) swap(u, v);
        add_segtree2_edge(1, 1, n, pos[head[u]], pos[v], node);
        u = par[head[u]];
    }
    if (depth[u] > depth[v]) swap(u, v);
    add_segtree2_edge(1, 1, n, pos[u], pos[v], node);
}

void solve() {
    init();
    cin >> n;
    for (int i = 0; i < n-1; i++) {
        int u, v; cin >> u >> v;
        g[u].emplace_back(v);
        g[v].emplace_back(u);
    } 
    cin >> m;
    vector<pair<int, int>> edges(m+1);
    for (int i = 1; i <= m; i++) {
        cin >> edges[i].first >> edges[i].second;
        auto [u, v] = edges[i];
        s[u] = i;
        e[v] = i;
    }
    build_segtree1(1, 1, n);
    build_segtree2(1, 1, n);
    dfs(1);
    dfs2(1, 1, -1);
    for (int i = 1; i <= m; i++) {
        auto [u, v] = edges[i];
        qry1(nxt(u, v), v, get_path_index(i));
        qry2(u, nxt(v, u), get_path_index(i));
    }
    for (int i = 1; i <= m; i++) {
        int u = get_path_index(i);
        if (col[u] == 0) dfs3(u);
    }
    cout << (ok?"Yes\n":"No\n");
}

int main() {
    ios::sync_with_stdio(0); cin.tie(0);
    int t; cin >> t;
    while (t--) {
        solve();
    }
}

Compilation message

jail.cpp: In function 'int nxt(int, int)':
jail.cpp:68:1: warning: control reaches end of non-void function [-Wreturn-type]
   68 | }
      | ^

Subtask #1
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 18 ms 113500 KB Output is correct
2 Correct 17 ms 113496 KB Output is correct
3 Correct 18 ms 113480 KB Output is correct
4 Correct 28 ms 113616 KB Output is correct
5 Correct 36 ms 113480 KB Output is correct
6 Correct 19 ms 113500 KB Output is correct
7 Correct 21 ms 113500 KB Output is correct
8 Correct 20 ms 113500 KB Output is correct
9 Correct 48 ms 115380 KB Output is correct
10 Correct 69 ms 139088 KB Output is correct
11 Correct 23 ms 113500 KB Output is correct
12 Correct 48 ms 113500 KB Output is correct
13 Correct 126 ms 147028 KB Output is correct
14 Correct 149 ms 147100 KB Output is correct
15 Correct 216 ms 152148 KB Output is correct
16 Correct 405 ms 175696 KB Output is correct
17 Correct 156 ms 154964 KB Output is correct
18 Correct 128 ms 150352 KB Output is correct
19 Correct 169 ms 153124 KB Output is correct
20 Correct 141 ms 152912 KB Output is correct
21 Correct 171 ms 154452 KB Output is correct
22 Correct 102 ms 144476 KB Output is correct

Subtask #2
0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 17 ms 113500 KB Output is correct
2 Correct 18 ms 113500 KB Output is correct
3 Correct 19 ms 113500 KB Output is correct
4 Incorrect 19 ms 113528 KB Output isn't correct
5 Halted 0 ms 0 KB -

Subtask #3
0 / 16.0

# Verdict Execution time Memory Grader output
1 Correct 17 ms 113500 KB Output is correct
2 Correct 18 ms 113500 KB Output is correct
3 Correct 19 ms 113500 KB Output is correct
4 Incorrect 19 ms 113528 KB Output isn't correct
5 Halted 0 ms 0 KB -

Subtask #4
0 / 28.0

# Verdict Execution time Memory Grader output
1 Correct 17 ms 113500 KB Output is correct
2 Correct 18 ms 113500 KB Output is correct
3 Correct 19 ms 113500 KB Output is correct
4 Incorrect 19 ms 113528 KB Output isn't correct
5 Halted 0 ms 0 KB -

Subtask #5
0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 17 ms 113500 KB Output is correct
2 Correct 18 ms 113500 KB Output is correct
3 Correct 19 ms 113500 KB Output is correct
4 Incorrect 19 ms 113528 KB Output isn't correct
5 Halted 0 ms 0 KB -

Subtask #6
0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 18 ms 113496 KB Output is correct
2 Correct 17 ms 113500 KB Output is correct
3 Correct 18 ms 113456 KB Output is correct
4 Correct 19 ms 113500 KB Output is correct
5 Correct 24 ms 113584 KB Output is correct
6 Correct 19 ms 113620 KB Output is correct
7 Incorrect 20 ms 113500 KB Output isn't correct
8 Halted 0 ms 0 KB -

Subtask #7
0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 18 ms 113500 KB Output is correct
2 Correct 17 ms 113496 KB Output is correct
3 Correct 18 ms 113480 KB Output is correct
4 Correct 28 ms 113616 KB Output is correct
5 Correct 36 ms 113480 KB Output is correct
6 Correct 19 ms 113500 KB Output is correct
7 Correct 21 ms 113500 KB Output is correct
8 Correct 20 ms 113500 KB Output is correct
9 Correct 48 ms 115380 KB Output is correct
10 Correct 69 ms 139088 KB Output is correct
11 Correct 23 ms 113500 KB Output is correct
12 Correct 48 ms 113500 KB Output is correct
13 Correct 126 ms 147028 KB Output is correct
14 Correct 149 ms 147100 KB Output is correct
15 Correct 216 ms 152148 KB Output is correct
16 Correct 405 ms 175696 KB Output is correct
17 Correct 156 ms 154964 KB Output is correct
18 Correct 128 ms 150352 KB Output is correct
19 Correct 169 ms 153124 KB Output is correct
20 Correct 141 ms 152912 KB Output is correct
21 Correct 171 ms 154452 KB Output is correct
22 Correct 102 ms 144476 KB Output is correct
23 Correct 17 ms 113500 KB Output is correct
24 Correct 18 ms 113500 KB Output is correct
25 Correct 19 ms 113500 KB Output is correct
26 Incorrect 19 ms 113528 KB Output isn't correct
27 Halted 0 ms 0 KB -