oj.uz

Submission #853677

# Submission time Handle Problem Language Result Execution time Memory
853677 2023-09-25T02:05:12 Z anha3k25cvp Birthday gift (IZhO18_treearray) C++14
56 / 100
 904 ms 84204 KB 
#include <bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define dl double
#define st first
#define nd second
#define II pair <int, int>

using namespace std;

const int N = 5 + 1e5;
const int inf = 7 + 1e9;

int node = 0, logn;
vector <int> num, h;
vector <vector <int>> adj, p;

void dfs(int u) {
    num[u] = ++ node;
    for (int v : adj[u])
        if (!num[v]) {
            p[0][v] = u;
            h[v] = h[u] + 1;
            dfs(v);
        }
}

int lca(int u, int v) {
    if (h[u] < h[v])
        swap(u, v);
    for (int si = h[u] - h[v]; si > 0; si ^= si & -si) {
        int i = __builtin_ctz(si & -si);
        u = p[i][u];
    }
    if (u == v)
        return u;
    for (int i = logn - 1; i >= 0; i --)
        if (p[i][u] != p[i][v]) {
            u = p[i][u];
            v = p[i][v];
        }
    return p[0][u];
}

int main() {
#define TASKNAME ""
    ios_base :: sync_with_stdio (0);
    cin.tie (0);
    if ( fopen( TASKNAME".inp", "r" ) ) {
        freopen (TASKNAME".inp", "r", stdin);
        freopen (TASKNAME".out", "w", stdout);
    }
    int n, m, t;
    cin >> n >> m >> t;
    adj.resize(n + 1);
    for (int i = 1; i < n; i ++) {
        int u, v;
        cin >> u >> v;
        adj[u].push_back(v);
        adj[v].push_back(u);
    }
    for (logn = 1; 1 << logn <= n; logn ++);
    p.assign(logn, vector <int> (n + 1, 0));
    num.assign(n + 1, 0);
    h.assign(n + 1, 0);
    h[1] = 1;
    dfs(1);
    for (int i = 1; i < logn; i ++)
        for (int u = 1; u <= n; u ++)
            p[i][u] = p[i - 1][p[i - 1][u]];
    vector <int> a(m + 1, 0), b(m, 0);
    vector <set <int>> q(n + 1), g(n + 1);
    for (int i = 1; i <= m; i ++) {
        cin >> a[i];
        g[a[i]].insert(i);
        if (i > 1) {
            b[i - 1] = num[lca(a[i - 1], a[i])];
            q[b[i - 1]].insert(i - 1);
        }
    }
    while (t --) {
        int type;
        cin >> type;
        if (type == 1) {
            int pos, val;
            cin >> pos >> val;
            g[a[pos]].erase(pos);
            a[pos] = val;
            g[val].insert(pos);
            if (pos > 1) {
                q[b[pos - 1]].erase(pos - 1);
                b[pos - 1] = num[lca(a[pos - 1], a[pos])];
                q[b[pos - 1]].insert(pos - 1);
            }
            if (pos < n) {
                q[b[pos]].erase(pos);
                b[pos] = num[lca(a[pos], a[pos + 1])];
                q[b[pos]].insert(pos);
            }
        }
        else {
            int u, v, val;
            cin >> u >> v >> val;
            auto it = q[num[val]].lower_bound(u), it1 = g[val].lower_bound(u);
            if (it != q[num[val]].end() &&  *it < v)
                cout << *it << ' ' << *it + 1 << '\n';
            else if (it1 != g[val].end() && *it1 <= v)
                cout << *it1 << ' ' << *it1 << '\n';
            else
                cout << "-1 -1\n";
        }
    }
    return 0;
}

Compilation message

treearray.cpp: In function 'int main()':
treearray.cpp:50:17: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   50 |         freopen (TASKNAME".inp", "r", stdin);
      |         ~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~
treearray.cpp:51:17: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   51 |         freopen (TASKNAME".out", "w", stdout);
      |         ~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~

Subtask #1
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 504 KB n=5
2 Correct 1 ms 600 KB n=100
3 Correct 0 ms 348 KB n=100
4 Correct 0 ms 344 KB n=100
5 Correct 1 ms 348 KB n=100
6 Correct 0 ms 348 KB n=100
7 Correct 0 ms 348 KB n=100
8 Correct 0 ms 348 KB n=100
9 Correct 0 ms 348 KB n=100
10 Correct 0 ms 460 KB n=100
11 Correct 1 ms 344 KB n=100
12 Correct 0 ms 348 KB n=100
13 Correct 0 ms 348 KB n=100
14 Correct 0 ms 344 KB n=100
15 Correct 1 ms 344 KB n=100
16 Correct 1 ms 348 KB n=100
17 Correct 0 ms 348 KB n=100
18 Correct 0 ms 344 KB n=100
19 Correct 1 ms 344 KB n=100
20 Correct 1 ms 344 KB n=100
21 Correct 0 ms 348 KB n=100
22 Correct 0 ms 348 KB n=100
23 Correct 0 ms 600 KB n=100
24 Correct 1 ms 344 KB n=100
25 Correct 0 ms 348 KB n=100
26 Correct 0 ms 344 KB n=12
27 Correct 0 ms 348 KB n=100

Subtask #2
18.0 / 18.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 504 KB n=5
2 Correct 1 ms 600 KB n=100
3 Correct 0 ms 348 KB n=100
4 Correct 0 ms 344 KB n=100
5 Correct 1 ms 348 KB n=100
6 Correct 0 ms 348 KB n=100
7 Correct 0 ms 348 KB n=100
8 Correct 0 ms 348 KB n=100
9 Correct 0 ms 348 KB n=100
10 Correct 0 ms 460 KB n=100
11 Correct 1 ms 344 KB n=100
12 Correct 0 ms 348 KB n=100
13 Correct 0 ms 348 KB n=100
14 Correct 0 ms 344 KB n=100
15 Correct 1 ms 344 KB n=100
16 Correct 1 ms 348 KB n=100
17 Correct 0 ms 348 KB n=100
18 Correct 0 ms 344 KB n=100
19 Correct 1 ms 344 KB n=100
20 Correct 1 ms 344 KB n=100
21 Correct 0 ms 348 KB n=100
22 Correct 0 ms 348 KB n=100
23 Correct 0 ms 600 KB n=100
24 Correct 1 ms 344 KB n=100
25 Correct 0 ms 348 KB n=100
26 Correct 0 ms 344 KB n=12
27 Correct 0 ms 348 KB n=100
28 Correct 1 ms 600 KB n=500
29 Correct 1 ms 604 KB n=500
30 Correct 1 ms 604 KB n=500
31 Correct 1 ms 604 KB n=500
32 Correct 1 ms 1012 KB n=500
33 Correct 1 ms 604 KB n=500
34 Correct 1 ms 604 KB n=500
35 Correct 1 ms 604 KB n=500
36 Correct 1 ms 604 KB n=500
37 Correct 1 ms 604 KB n=500
38 Correct 1 ms 604 KB n=500
39 Correct 1 ms 604 KB n=500
40 Correct 1 ms 604 KB n=500
41 Correct 1 ms 604 KB n=500
42 Correct 1 ms 604 KB n=500
43 Correct 1 ms 604 KB n=500
44 Correct 1 ms 604 KB n=500
45 Correct 1 ms 604 KB n=500
46 Correct 1 ms 604 KB n=500
47 Correct 1 ms 604 KB n=500
48 Correct 1 ms 604 KB n=500
49 Correct 1 ms 604 KB n=500
50 Correct 1 ms 604 KB n=500
51 Correct 1 ms 604 KB n=500
52 Correct 1 ms 604 KB n=500
53 Correct 1 ms 604 KB n=500
54 Correct 1 ms 604 KB n=500
55 Correct 1 ms 348 KB n=278
56 Correct 1 ms 600 KB n=500
57 Correct 1 ms 604 KB n=500
58 Correct 1 ms 604 KB n=500

Subtask #3
26.0 / 26.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 504 KB n=5
2 Correct 1 ms 600 KB n=100
3 Correct 0 ms 348 KB n=100
4 Correct 0 ms 344 KB n=100
5 Correct 1 ms 348 KB n=100
6 Correct 0 ms 348 KB n=100
7 Correct 0 ms 348 KB n=100
8 Correct 0 ms 348 KB n=100
9 Correct 0 ms 348 KB n=100
10 Correct 0 ms 460 KB n=100
11 Correct 1 ms 344 KB n=100
12 Correct 0 ms 348 KB n=100
13 Correct 0 ms 348 KB n=100
14 Correct 0 ms 344 KB n=100
15 Correct 1 ms 344 KB n=100
16 Correct 1 ms 348 KB n=100
17 Correct 0 ms 348 KB n=100
18 Correct 0 ms 344 KB n=100
19 Correct 1 ms 344 KB n=100
20 Correct 1 ms 344 KB n=100
21 Correct 0 ms 348 KB n=100
22 Correct 0 ms 348 KB n=100
23 Correct 0 ms 600 KB n=100
24 Correct 1 ms 344 KB n=100
25 Correct 0 ms 348 KB n=100
26 Correct 0 ms 344 KB n=12
27 Correct 0 ms 348 KB n=100
28 Correct 1 ms 600 KB n=500
29 Correct 1 ms 604 KB n=500
30 Correct 1 ms 604 KB n=500
31 Correct 1 ms 604 KB n=500
32 Correct 1 ms 1012 KB n=500
33 Correct 1 ms 604 KB n=500
34 Correct 1 ms 604 KB n=500
35 Correct 1 ms 604 KB n=500
36 Correct 1 ms 604 KB n=500
37 Correct 1 ms 604 KB n=500
38 Correct 1 ms 604 KB n=500
39 Correct 1 ms 604 KB n=500
40 Correct 1 ms 604 KB n=500
41 Correct 1 ms 604 KB n=500
42 Correct 1 ms 604 KB n=500
43 Correct 1 ms 604 KB n=500
44 Correct 1 ms 604 KB n=500
45 Correct 1 ms 604 KB n=500
46 Correct 1 ms 604 KB n=500
47 Correct 1 ms 604 KB n=500
48 Correct 1 ms 604 KB n=500
49 Correct 1 ms 604 KB n=500
50 Correct 1 ms 604 KB n=500
51 Correct 1 ms 604 KB n=500
52 Correct 1 ms 604 KB n=500
53 Correct 1 ms 604 KB n=500
54 Correct 1 ms 604 KB n=500
55 Correct 1 ms 348 KB n=278
56 Correct 1 ms 600 KB n=500
57 Correct 1 ms 604 KB n=500
58 Correct 1 ms 604 KB n=500
59 Correct 2 ms 1116 KB n=2000
60 Correct 2 ms 1112 KB n=2000
61 Correct 3 ms 1116 KB n=2000
62 Correct 3 ms 1112 KB n=2000
63 Correct 2 ms 1116 KB n=2000
64 Correct 3 ms 1116 KB n=2000
65 Correct 2 ms 1116 KB n=2000
66 Correct 2 ms 1116 KB n=2000
67 Correct 3 ms 1116 KB n=2000
68 Correct 2 ms 976 KB n=2000
69 Correct 2 ms 1116 KB n=2000
70 Correct 3 ms 1116 KB n=2000
71 Correct 2 ms 1116 KB n=2000
72 Correct 2 ms 1116 KB n=2000
73 Correct 2 ms 1112 KB n=2000
74 Correct 2 ms 860 KB n=1844
75 Correct 2 ms 1112 KB n=2000
76 Correct 3 ms 1112 KB n=2000
77 Correct 3 ms 1112 KB n=2000
78 Correct 3 ms 1116 KB n=2000
79 Correct 5 ms 1116 KB n=2000
80 Correct 3 ms 1116 KB n=2000
81 Correct 2 ms 1116 KB n=2000
82 Correct 3 ms 1116 KB n=2000
83 Correct 2 ms 1116 KB n=2000
84 Correct 3 ms 1116 KB n=2000
85 Correct 3 ms 1116 KB n=2000
86 Correct 2 ms 1112 KB n=2000
87 Correct 2 ms 1116 KB n=2000
88 Correct 2 ms 1116 KB n=2000
89 Correct 2 ms 1116 KB n=2000
90 Correct 2 ms 1116 KB n=2000
91 Correct 2 ms 1116 KB n=2000

Subtask #4
0 / 44.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 504 KB n=5
2 Correct 1 ms 600 KB n=100
3 Correct 0 ms 348 KB n=100
4 Correct 0 ms 344 KB n=100
5 Correct 1 ms 348 KB n=100
6 Correct 0 ms 348 KB n=100
7 Correct 0 ms 348 KB n=100
8 Correct 0 ms 348 KB n=100
9 Correct 0 ms 348 KB n=100
10 Correct 0 ms 460 KB n=100
11 Correct 1 ms 344 KB n=100
12 Correct 0 ms 348 KB n=100
13 Correct 0 ms 348 KB n=100
14 Correct 0 ms 344 KB n=100
15 Correct 1 ms 344 KB n=100
16 Correct 1 ms 348 KB n=100
17 Correct 0 ms 348 KB n=100
18 Correct 0 ms 344 KB n=100
19 Correct 1 ms 344 KB n=100
20 Correct 1 ms 344 KB n=100
21 Correct 0 ms 348 KB n=100
22 Correct 0 ms 348 KB n=100
23 Correct 0 ms 600 KB n=100
24 Correct 1 ms 344 KB n=100
25 Correct 0 ms 348 KB n=100
26 Correct 0 ms 344 KB n=12
27 Correct 0 ms 348 KB n=100
28 Correct 1 ms 600 KB n=500
29 Correct 1 ms 604 KB n=500
30 Correct 1 ms 604 KB n=500
31 Correct 1 ms 604 KB n=500
32 Correct 1 ms 1012 KB n=500
33 Correct 1 ms 604 KB n=500
34 Correct 1 ms 604 KB n=500
35 Correct 1 ms 604 KB n=500
36 Correct 1 ms 604 KB n=500
37 Correct 1 ms 604 KB n=500
38 Correct 1 ms 604 KB n=500
39 Correct 1 ms 604 KB n=500
40 Correct 1 ms 604 KB n=500
41 Correct 1 ms 604 KB n=500
42 Correct 1 ms 604 KB n=500
43 Correct 1 ms 604 KB n=500
44 Correct 1 ms 604 KB n=500
45 Correct 1 ms 604 KB n=500
46 Correct 1 ms 604 KB n=500
47 Correct 1 ms 604 KB n=500
48 Correct 1 ms 604 KB n=500
49 Correct 1 ms 604 KB n=500
50 Correct 1 ms 604 KB n=500
51 Correct 1 ms 604 KB n=500
52 Correct 1 ms 604 KB n=500
53 Correct 1 ms 604 KB n=500
54 Correct 1 ms 604 KB n=500
55 Correct 1 ms 348 KB n=278
56 Correct 1 ms 600 KB n=500
57 Correct 1 ms 604 KB n=500
58 Correct 1 ms 604 KB n=500
59 Correct 2 ms 1116 KB n=2000
60 Correct 2 ms 1112 KB n=2000
61 Correct 3 ms 1116 KB n=2000
62 Correct 3 ms 1112 KB n=2000
63 Correct 2 ms 1116 KB n=2000
64 Correct 3 ms 1116 KB n=2000
65 Correct 2 ms 1116 KB n=2000
66 Correct 2 ms 1116 KB n=2000
67 Correct 3 ms 1116 KB n=2000
68 Correct 2 ms 976 KB n=2000
69 Correct 2 ms 1116 KB n=2000
70 Correct 3 ms 1116 KB n=2000
71 Correct 2 ms 1116 KB n=2000
72 Correct 2 ms 1116 KB n=2000
73 Correct 2 ms 1112 KB n=2000
74 Correct 2 ms 860 KB n=1844
75 Correct 2 ms 1112 KB n=2000
76 Correct 3 ms 1112 KB n=2000
77 Correct 3 ms 1112 KB n=2000
78 Correct 3 ms 1116 KB n=2000
79 Correct 5 ms 1116 KB n=2000
80 Correct 3 ms 1116 KB n=2000
81 Correct 2 ms 1116 KB n=2000
82 Correct 3 ms 1116 KB n=2000
83 Correct 2 ms 1116 KB n=2000
84 Correct 3 ms 1116 KB n=2000
85 Correct 3 ms 1116 KB n=2000
86 Correct 2 ms 1112 KB n=2000
87 Correct 2 ms 1116 KB n=2000
88 Correct 2 ms 1116 KB n=2000
89 Correct 2 ms 1116 KB n=2000
90 Correct 2 ms 1116 KB n=2000
91 Correct 2 ms 1116 KB n=2000
92 Correct 555 ms 74684 KB n=200000
93 Correct 694 ms 79684 KB n=200000
94 Correct 716 ms 83136 KB n=200000
95 Correct 567 ms 74376 KB n=200000
96 Correct 560 ms 74328 KB n=200000
97 Correct 830 ms 78676 KB n=200000
98 Correct 549 ms 74516 KB n=200000
99 Correct 759 ms 74532 KB n=200000
100 Correct 551 ms 74532 KB n=200000
101 Correct 667 ms 84204 KB n=200000
102 Correct 335 ms 75576 KB n=200000
103 Correct 346 ms 75588 KB n=200000
104 Correct 345 ms 75632 KB n=200000
105 Correct 361 ms 76104 KB n=200000
106 Correct 358 ms 76096 KB n=200000
107 Correct 373 ms 75980 KB n=200000
108 Correct 681 ms 74884 KB n=200000
109 Correct 639 ms 74744 KB n=200000
110 Correct 672 ms 74640 KB n=200000
111 Correct 579 ms 74008 KB n=200000
112 Correct 634 ms 83116 KB n=200000
113 Correct 782 ms 78580 KB n=200000
114 Correct 568 ms 73868 KB n=200000
115 Correct 904 ms 76472 KB n=200000
116 Correct 537 ms 74728 KB n=200000
117 Correct 622 ms 83500 KB n=200000
118 Correct 810 ms 77488 KB n=200000
119 Correct 505 ms 74564 KB n=200000
120 Correct 598 ms 83060 KB n=200000
121 Correct 631 ms 82984 KB n=200000
122 Correct 641 ms 83472 KB n=200000
123 Correct 373 ms 75732 KB n=200000
124 Incorrect 91 ms 18260 KB Wrong range.
125 Halted 0 ms 0 KB -