oj.uz

답안 #879285

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
879285 2023-11-27T05:28:21 Z The_Samurai Birthday gift (IZhO18_treearray) C++17
56 / 100
 442 ms 262144 KB 
// I stand with PALESTINE




//#pragma GCC optimize("Ofast,O3")
//#pragma GCC target("avx,avx2")
#include "bits/stdc++.h"

using namespace std;
using ll = long long;
const int lg = 18;

vector<vector<int>> g, jump;
vector<int> tin, tout;
int z;

void dfs(int u) {
    if (u != 1) {
        for (int i = 1; i < lg; i++) {
            jump[i][u] = jump[i - 1][jump[i - 1][u]];
            if (!jump[i][u]) break;
        }
    }
    tin[u] = z++;
    for (int v: g[u]) {
        if (jump[0][u] == v) continue;
        jump[0][v] = u;
        dfs(v);
    }
    tout[u] = z++;
}

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

int lca(int u, int v) {
    if (v == 0 or isPar(u, v)) return u;
    if (u == 0 or isPar(v, u)) return v;
    for (int i = lg - 1; i >= 0; i--) {
        if (!jump[i][u] or isPar(jump[i][u], v)) continue;
        u = jump[i][u];
    }
    return jump[0][u];
}

struct SegTree {
    vector<set<pair<int, int>>> tree;
    int size;

    void init(int n) {
        size = 1;
        while (size < n) size *= 2;
        tree.assign(2 * size - 1, set<pair<int, int>>());
        build(0, 0, size);
    }

    void build(int x, int lx, int rx) {
        for (int i = 0; i < rx - lx; i++) tree[x].emplace(0, i + lx);
        if (rx - lx == 1) return;
        int m = (lx + rx) >> 1;
        build(2 * x + 1, lx, m);
        build(2 * x + 2, m, rx);
    }

    int upd(int i, int v, int x, int lx, int rx) {
        if (rx - lx == 1) {
            int vv = tree[x].begin()->first;
            tree[x] = set<pair<int, int>>{make_pair(v, i)};
            return vv;
        }
        int vv, m = (lx + rx) >> 1;
        if (i < m) vv = upd(i, v, 2 * x + 1, lx, m);
        else vv = upd(i, v, 2 * x + 2, m, rx);
        tree[x].erase(tree[x].find(make_pair(vv, i)));
        tree[x].emplace(v, i);
        return vv;
    }

    void upd(int i, int v) {
        upd(i, v, 0, 0, size);
    }

    int get(int l, int r, int v, int x, int lx, int rx) {
        if (l >= rx or lx >= r) return -1;
        if (l <= lx and rx <= r) {
            auto it = tree[x].lower_bound(make_pair(v, 0));
            if (it != tree[x].end() and it->first == v) return it->second;
            return -1;
        }
        int m = (lx + rx) >> 1;
        int ans = get(l, r, v, 2 * x + 1, lx, m);
        if (ans != -1) return ans;
        ans = get(l, r, v, 2 * x + 2, m, rx);
        return ans;
    }

    int get(int l, int r, int v) {
        return get(l, r + 1, v, 0, 0, size);
    }
};

void solve() {
    int n, m, q;
    cin >> n >> m >> q;
    g.assign(n + 1, {});
    jump.assign(lg, vector<int>(n + 1));
    tin.assign(n + 1, 0); tout.assign(n + 1, 0);
    for (int i = 1; i < n; i++) {
        int u, v;
        cin >> u >> v;
        g[u].emplace_back(v);
        g[v].emplace_back(u);
    }
    vector<int> a(m + 1);
    for (int i = 1; i <= m; i++) cin >> a[i];
    dfs(1);
    SegTree sg1, sg2;
    sg1.init(m + 1); sg2.init(m + 1);
    for (int i = 1; i <= m; i++) sg1.upd(i, a[i]);
    for (int i = 1; i < m; i++) sg2.upd(i, lca(a[i], a[i + 1]));
    while (q--) {
        int op;
        cin >> op;
        if (op == 1) {
            int i;
            cin >> i >> a[i];
            sg1.upd(i, a[i]);
            if (i > 1) sg2.upd(i - 1, lca(a[i - 1], a[i]));
            if (i < m) sg2.upd(i, lca(a[i], a[i + 1]));
        } else {
            int l, r, root;
            cin >> l >> r >> root;
            int ans = sg1.get(l, r, root);
            if (ans != -1) {
                cout << ans << ' ' << ans << '\n';
                continue;
            }
            ans = sg2.get(l, r - 1, root);
            if (ans != -1) cout << ans << ' ' << ans + 1 << '\n';
            else cout << "-1 -1\n";
        }
    }
}

int main() {
    cin.tie(0)->sync_with_stdio(false);
#ifdef sunnatov
    freopen("input.txt", "r", stdin);
    freopen("output.txt", "w", stdout);
#endif
    int q = 1;
//    cin >> q;
    while (q--) {
        solve();
        cout << '\n';
    }
}

Subtask #1
12.0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 348 KB n=5
2 Correct 1 ms 348 KB n=100
3 Correct 1 ms 348 KB n=100
4 Correct 1 ms 348 KB n=100
5 Correct 1 ms 348 KB n=100
6 Correct 1 ms 344 KB n=100
7 Correct 1 ms 348 KB n=100
8 Correct 1 ms 388 KB n=100
9 Correct 1 ms 348 KB n=100
10 Correct 1 ms 348 KB n=100
11 Correct 1 ms 348 KB n=100
12 Correct 1 ms 348 KB n=100
13 Correct 1 ms 348 KB n=100
14 Correct 1 ms 344 KB n=100
15 Correct 1 ms 344 KB n=100
16 Correct 1 ms 344 KB n=100
17 Correct 1 ms 344 KB n=100
18 Correct 1 ms 344 KB n=100
19 Correct 1 ms 348 KB n=100
20 Correct 1 ms 348 KB n=100
21 Correct 1 ms 348 KB n=100
22 Correct 1 ms 348 KB n=100
23 Correct 1 ms 348 KB n=100
24 Correct 1 ms 348 KB n=100
25 Correct 1 ms 348 KB n=100
26 Correct 1 ms 344 KB n=12
27 Correct 1 ms 344 KB n=100

Subtask #2
18.0 / 18.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 348 KB n=5
2 Correct 1 ms 348 KB n=100
3 Correct 1 ms 348 KB n=100
4 Correct 1 ms 348 KB n=100
5 Correct 1 ms 348 KB n=100
6 Correct 1 ms 344 KB n=100
7 Correct 1 ms 348 KB n=100
8 Correct 1 ms 388 KB n=100
9 Correct 1 ms 348 KB n=100
10 Correct 1 ms 348 KB n=100
11 Correct 1 ms 348 KB n=100
12 Correct 1 ms 348 KB n=100
13 Correct 1 ms 348 KB n=100
14 Correct 1 ms 344 KB n=100
15 Correct 1 ms 344 KB n=100
16 Correct 1 ms 344 KB n=100
17 Correct 1 ms 344 KB n=100
18 Correct 1 ms 344 KB n=100
19 Correct 1 ms 348 KB n=100
20 Correct 1 ms 348 KB n=100
21 Correct 1 ms 348 KB n=100
22 Correct 1 ms 348 KB n=100
23 Correct 1 ms 348 KB n=100
24 Correct 1 ms 348 KB n=100
25 Correct 1 ms 348 KB n=100
26 Correct 1 ms 344 KB n=12
27 Correct 1 ms 344 KB n=100
28 Correct 4 ms 856 KB n=500
29 Correct 3 ms 1116 KB n=500
30 Correct 4 ms 856 KB n=500
31 Correct 4 ms 1116 KB n=500
32 Correct 3 ms 856 KB n=500
33 Correct 3 ms 860 KB n=500
34 Correct 3 ms 860 KB n=500
35 Correct 3 ms 1116 KB n=500
36 Correct 2 ms 860 KB n=500
37 Correct 2 ms 1112 KB n=500
38 Correct 2 ms 860 KB n=500
39 Correct 2 ms 860 KB n=500
40 Correct 3 ms 860 KB n=500
41 Correct 2 ms 856 KB n=500
42 Correct 3 ms 860 KB n=500
43 Correct 3 ms 860 KB n=500
44 Correct 3 ms 860 KB n=500
45 Correct 3 ms 860 KB n=500
46 Correct 3 ms 1116 KB n=500
47 Correct 3 ms 1116 KB n=500
48 Correct 3 ms 860 KB n=500
49 Correct 3 ms 1116 KB n=500
50 Correct 4 ms 860 KB n=500
51 Correct 3 ms 860 KB n=500
52 Correct 4 ms 1116 KB n=500
53 Correct 4 ms 860 KB n=500
54 Correct 4 ms 1116 KB n=500
55 Correct 3 ms 860 KB n=278
56 Correct 4 ms 1116 KB n=500
57 Correct 4 ms 1116 KB n=500
58 Correct 2 ms 856 KB n=500

Subtask #3
26.0 / 26.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 348 KB n=5
2 Correct 1 ms 348 KB n=100
3 Correct 1 ms 348 KB n=100
4 Correct 1 ms 348 KB n=100
5 Correct 1 ms 348 KB n=100
6 Correct 1 ms 344 KB n=100
7 Correct 1 ms 348 KB n=100
8 Correct 1 ms 388 KB n=100
9 Correct 1 ms 348 KB n=100
10 Correct 1 ms 348 KB n=100
11 Correct 1 ms 348 KB n=100
12 Correct 1 ms 348 KB n=100
13 Correct 1 ms 348 KB n=100
14 Correct 1 ms 344 KB n=100
15 Correct 1 ms 344 KB n=100
16 Correct 1 ms 344 KB n=100
17 Correct 1 ms 344 KB n=100
18 Correct 1 ms 344 KB n=100
19 Correct 1 ms 348 KB n=100
20 Correct 1 ms 348 KB n=100
21 Correct 1 ms 348 KB n=100
22 Correct 1 ms 348 KB n=100
23 Correct 1 ms 348 KB n=100
24 Correct 1 ms 348 KB n=100
25 Correct 1 ms 348 KB n=100
26 Correct 1 ms 344 KB n=12
27 Correct 1 ms 344 KB n=100
28 Correct 4 ms 856 KB n=500
29 Correct 3 ms 1116 KB n=500
30 Correct 4 ms 856 KB n=500
31 Correct 4 ms 1116 KB n=500
32 Correct 3 ms 856 KB n=500
33 Correct 3 ms 860 KB n=500
34 Correct 3 ms 860 KB n=500
35 Correct 3 ms 1116 KB n=500
36 Correct 2 ms 860 KB n=500
37 Correct 2 ms 1112 KB n=500
38 Correct 2 ms 860 KB n=500
39 Correct 2 ms 860 KB n=500
40 Correct 3 ms 860 KB n=500
41 Correct 2 ms 856 KB n=500
42 Correct 3 ms 860 KB n=500
43 Correct 3 ms 860 KB n=500
44 Correct 3 ms 860 KB n=500
45 Correct 3 ms 860 KB n=500
46 Correct 3 ms 1116 KB n=500
47 Correct 3 ms 1116 KB n=500
48 Correct 3 ms 860 KB n=500
49 Correct 3 ms 1116 KB n=500
50 Correct 4 ms 860 KB n=500
51 Correct 3 ms 860 KB n=500
52 Correct 4 ms 1116 KB n=500
53 Correct 4 ms 860 KB n=500
54 Correct 4 ms 1116 KB n=500
55 Correct 3 ms 860 KB n=278
56 Correct 4 ms 1116 KB n=500
57 Correct 4 ms 1116 KB n=500
58 Correct 2 ms 856 KB n=500
59 Correct 18 ms 3416 KB n=2000
60 Correct 19 ms 3420 KB n=2000
61 Correct 19 ms 3416 KB n=2000
62 Correct 19 ms 3416 KB n=2000
63 Correct 19 ms 3456 KB n=2000
64 Correct 20 ms 3420 KB n=2000
65 Correct 17 ms 3420 KB n=2000
66 Correct 18 ms 3420 KB n=2000
67 Correct 19 ms 3440 KB n=2000
68 Correct 20 ms 3616 KB n=2000
69 Correct 11 ms 3420 KB n=2000
70 Correct 12 ms 3416 KB n=2000
71 Correct 11 ms 3420 KB n=2000
72 Correct 12 ms 3420 KB n=2000
73 Correct 11 ms 3416 KB n=2000
74 Correct 16 ms 3420 KB n=1844
75 Correct 12 ms 3416 KB n=2000
76 Correct 18 ms 3416 KB n=2000
77 Correct 18 ms 3436 KB n=2000
78 Correct 17 ms 3416 KB n=2000
79 Correct 17 ms 3420 KB n=2000
80 Correct 19 ms 3416 KB n=2000
81 Correct 19 ms 3420 KB n=2000
82 Correct 17 ms 3420 KB n=2000
83 Correct 19 ms 3420 KB n=2000
84 Correct 18 ms 3448 KB n=2000
85 Correct 19 ms 3484 KB n=2000
86 Correct 20 ms 3480 KB n=2000
87 Correct 18 ms 3420 KB n=2000
88 Correct 19 ms 3420 KB n=2000
89 Correct 19 ms 3416 KB n=2000
90 Correct 18 ms 3420 KB n=2000
91 Correct 13 ms 3672 KB n=2000

Subtask #4
0 / 44.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 348 KB n=5
2 Correct 1 ms 348 KB n=100
3 Correct 1 ms 348 KB n=100
4 Correct 1 ms 348 KB n=100
5 Correct 1 ms 348 KB n=100
6 Correct 1 ms 344 KB n=100
7 Correct 1 ms 348 KB n=100
8 Correct 1 ms 388 KB n=100
9 Correct 1 ms 348 KB n=100
10 Correct 1 ms 348 KB n=100
11 Correct 1 ms 348 KB n=100
12 Correct 1 ms 348 KB n=100
13 Correct 1 ms 348 KB n=100
14 Correct 1 ms 344 KB n=100
15 Correct 1 ms 344 KB n=100
16 Correct 1 ms 344 KB n=100
17 Correct 1 ms 344 KB n=100
18 Correct 1 ms 344 KB n=100
19 Correct 1 ms 348 KB n=100
20 Correct 1 ms 348 KB n=100
21 Correct 1 ms 348 KB n=100
22 Correct 1 ms 348 KB n=100
23 Correct 1 ms 348 KB n=100
24 Correct 1 ms 348 KB n=100
25 Correct 1 ms 348 KB n=100
26 Correct 1 ms 344 KB n=12
27 Correct 1 ms 344 KB n=100
28 Correct 4 ms 856 KB n=500
29 Correct 3 ms 1116 KB n=500
30 Correct 4 ms 856 KB n=500
31 Correct 4 ms 1116 KB n=500
32 Correct 3 ms 856 KB n=500
33 Correct 3 ms 860 KB n=500
34 Correct 3 ms 860 KB n=500
35 Correct 3 ms 1116 KB n=500
36 Correct 2 ms 860 KB n=500
37 Correct 2 ms 1112 KB n=500
38 Correct 2 ms 860 KB n=500
39 Correct 2 ms 860 KB n=500
40 Correct 3 ms 860 KB n=500
41 Correct 2 ms 856 KB n=500
42 Correct 3 ms 860 KB n=500
43 Correct 3 ms 860 KB n=500
44 Correct 3 ms 860 KB n=500
45 Correct 3 ms 860 KB n=500
46 Correct 3 ms 1116 KB n=500
47 Correct 3 ms 1116 KB n=500
48 Correct 3 ms 860 KB n=500
49 Correct 3 ms 1116 KB n=500
50 Correct 4 ms 860 KB n=500
51 Correct 3 ms 860 KB n=500
52 Correct 4 ms 1116 KB n=500
53 Correct 4 ms 860 KB n=500
54 Correct 4 ms 1116 KB n=500
55 Correct 3 ms 860 KB n=278
56 Correct 4 ms 1116 KB n=500
57 Correct 4 ms 1116 KB n=500
58 Correct 2 ms 856 KB n=500
59 Correct 18 ms 3416 KB n=2000
60 Correct 19 ms 3420 KB n=2000
61 Correct 19 ms 3416 KB n=2000
62 Correct 19 ms 3416 KB n=2000
63 Correct 19 ms 3456 KB n=2000
64 Correct 20 ms 3420 KB n=2000
65 Correct 17 ms 3420 KB n=2000
66 Correct 18 ms 3420 KB n=2000
67 Correct 19 ms 3440 KB n=2000
68 Correct 20 ms 3616 KB n=2000
69 Correct 11 ms 3420 KB n=2000
70 Correct 12 ms 3416 KB n=2000
71 Correct 11 ms 3420 KB n=2000
72 Correct 12 ms 3420 KB n=2000
73 Correct 11 ms 3416 KB n=2000
74 Correct 16 ms 3420 KB n=1844
75 Correct 12 ms 3416 KB n=2000
76 Correct 18 ms 3416 KB n=2000
77 Correct 18 ms 3436 KB n=2000
78 Correct 17 ms 3416 KB n=2000
79 Correct 17 ms 3420 KB n=2000
80 Correct 19 ms 3416 KB n=2000
81 Correct 19 ms 3420 KB n=2000
82 Correct 17 ms 3420 KB n=2000
83 Correct 19 ms 3420 KB n=2000
84 Correct 18 ms 3448 KB n=2000
85 Correct 19 ms 3484 KB n=2000
86 Correct 20 ms 3480 KB n=2000
87 Correct 18 ms 3420 KB n=2000
88 Correct 19 ms 3420 KB n=2000
89 Correct 19 ms 3416 KB n=2000
90 Correct 18 ms 3420 KB n=2000
91 Correct 13 ms 3672 KB n=2000
92 Runtime error 442 ms 262144 KB Execution killed with signal 9
93 Halted 0 ms 0 KB -