oj.uz

Submission #489891

# Submission time Handle Problem Language Result Execution time Memory
489891 2021-11-25T02:55:20 Z KienTran Construction of Highway (JOI18_construction) C++14
100 / 100
 1507 ms 231768 KB 
#include <bits/stdc++.h>
#define int long long

using namespace std;

const int O = 3e5 + 5;

int n, cnt, a[O], b[O], c[O], h[O], d[O], E[O], par[O], head[O], child[O], tree[O * 4];
vector <int> val, g[O];

deque <pair <int, int>> q[O];

void Update(int id, int l, int r, int u, int x){
    if (l > u || r < u) return;
    if (l == r){
        tree[id] += x;
        return;
    }
    int mid = (l + r) / 2;
    Update(id << 1, l, mid, u, x);
    Update(id << 1 | 1, mid + 1, r, u, x);
    tree[id] = tree[id << 1] + tree[id << 1 | 1];
}

int Get(int id, int l, int r, int u, int v){
    if (l > v || r < u) return 0;
    if (u <= l && r <= v) return tree[id];
    int mid = (l + r) / 2;
    return Get(id << 1, l, mid, u, v) + Get(id << 1 | 1, mid + 1, r, u, v);
}

void dfs(int u){
    child[u] = 1;
    for (int i : g[u]){
        int v = u ^ E[i];
        if (v != par[u]){
            par[v] = u;
            h[v] = h[u] + 1;
            dfs(v);
            child[u] += child[v];
        }
    }
}

void hld(int u){
    if (head[cnt] == 0) head[cnt] = u; d[u] = cnt;

    int Max = -1, BigChild = -1;

    for (int i : g[u]){
        int v = u ^ E[i];
        if (v != par[u] && child[v] > Max){
            Max = child[v];
            BigChild = v;
        }
    }

    if (BigChild != -1) hld(BigChild);

    for (int i : g[u]){
        int v = u ^ E[i];
        if (v != par[u] && v != BigChild){
            cnt += 1;
            hld(v);
        }
    }
}

main(){
    ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0);
    cin >> n;
    for (int i = 1; i <= n; ++ i){
        cin >> c[i];
        val.push_back(c[i]);
    }

    sort(val.begin(), val.end());
    val.resize(unique(val.begin(), val.end()) - val.begin());
    int lim = val.size();

    for (int i = 1; i < n; ++ i){
        cin >> a[i] >> b[i];
        g[a[i]].push_back(i);
        E[i] = a[i] ^ b[i];
    }

    dfs(1); hld(1);

    q[d[1]].push_back({1, c[1]});
    for (int i = 1; i < n; ++ i){
        int u = a[i];

        vector <pair <int, int>> need;
        while (u){
            int parU = head[d[u]];
            int k = h[u] - h[parU] + 1;

            vector <pair <int, int>> sth;
            while (k){
                int cost = min(k, q[d[u]][0].first);
                k -= cost;

                sth.push_back({cost, q[d[u]][0].second});

                q[d[u]][0].first -= cost;
                if (q[d[u]][0].first == 0) q[d[u]].pop_front();
            }


            while (sth.size()){
                need.push_back(sth.back());
                sth.pop_back();
            }

            q[d[u]].push_front({h[u] - h[parU] + 1, c[b[i]]});
            u = par[parU];
        }

        u = b[i];
        if (q[d[u]].size() && q[d[u]].back().second == c[u]) q[d[u]].back().first += 1;
        else q[d[u]].push_back({1, c[u]});

        int res = 0;
        for (auto j : need){
            int id = upper_bound(val.begin(), val.end(), j.second) - val.begin();
            res += j.first * Get(1, 1, lim, 1, id - 1);
            Update(1, 1, lim, id, j.first);
        }

        for (auto j : need){
            int id = upper_bound(val.begin(), val.end(), j.second) - val.begin();
            Update(1, 1, lim, id, -j.first);
        }

        cout << res << "\n";
    }
}

Compilation message

construction.cpp: In function 'void hld(long long int)':
construction.cpp:46:5: warning: this 'if' clause does not guard... [-Wmisleading-indentation]
   46 |     if (head[cnt] == 0) head[cnt] = u; d[u] = cnt;
      |     ^~
construction.cpp:46:40: note: ...this statement, but the latter is misleadingly indented as if it were guarded by the 'if'
   46 |     if (head[cnt] == 0) head[cnt] = u; d[u] = cnt;
      |                                        ^
construction.cpp: At global scope:
construction.cpp:69:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   69 | main(){
      | ^~~~

Subtask #1
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 131 ms 209480 KB Output is correct
2 Correct 130 ms 209380 KB Output is correct
3 Correct 132 ms 209552 KB Output is correct
4 Correct 143 ms 209436 KB Output is correct
5 Correct 152 ms 209528 KB Output is correct
6 Correct 140 ms 209416 KB Output is correct
7 Correct 145 ms 209516 KB Output is correct
8 Correct 137 ms 209364 KB Output is correct
9 Correct 134 ms 209480 KB Output is correct
10 Correct 129 ms 209460 KB Output is correct
11 Correct 135 ms 209472 KB Output is correct
12 Correct 130 ms 209368 KB Output is correct
13 Correct 133 ms 209540 KB Output is correct
14 Correct 129 ms 209404 KB Output is correct
15 Correct 147 ms 209436 KB Output is correct
16 Correct 158 ms 209464 KB Output is correct
17 Correct 141 ms 209588 KB Output is correct
18 Correct 145 ms 209480 KB Output is correct
19 Correct 132 ms 209464 KB Output is correct
20 Correct 136 ms 209420 KB Output is correct
21 Correct 132 ms 209412 KB Output is correct
22 Correct 150 ms 209472 KB Output is correct
23 Correct 140 ms 209452 KB Output is correct
24 Correct 154 ms 209340 KB Output is correct
25 Correct 142 ms 209404 KB Output is correct
26 Correct 137 ms 209508 KB Output is correct
27 Correct 128 ms 209420 KB Output is correct
28 Correct 135 ms 209416 KB Output is correct
29 Correct 140 ms 209412 KB Output is correct
30 Correct 147 ms 209392 KB Output is correct
31 Correct 147 ms 209512 KB Output is correct
32 Correct 131 ms 209432 KB Output is correct
33 Correct 156 ms 209404 KB Output is correct
34 Correct 131 ms 209448 KB Output is correct
35 Correct 129 ms 209440 KB Output is correct
36 Correct 145 ms 209432 KB Output is correct
37 Correct 139 ms 209348 KB Output is correct
38 Correct 128 ms 209320 KB Output is correct
39 Correct 135 ms 209452 KB Output is correct
40 Correct 128 ms 209416 KB Output is correct
41 Correct 130 ms 209324 KB Output is correct
42 Correct 129 ms 209364 KB Output is correct
43 Correct 135 ms 209320 KB Output is correct
44 Correct 128 ms 209328 KB Output is correct

Subtask #2
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 131 ms 209480 KB Output is correct
2 Correct 130 ms 209380 KB Output is correct
3 Correct 132 ms 209552 KB Output is correct
4 Correct 143 ms 209436 KB Output is correct
5 Correct 152 ms 209528 KB Output is correct
6 Correct 140 ms 209416 KB Output is correct
7 Correct 145 ms 209516 KB Output is correct
8 Correct 137 ms 209364 KB Output is correct
9 Correct 134 ms 209480 KB Output is correct
10 Correct 129 ms 209460 KB Output is correct
11 Correct 135 ms 209472 KB Output is correct
12 Correct 130 ms 209368 KB Output is correct
13 Correct 133 ms 209540 KB Output is correct
14 Correct 129 ms 209404 KB Output is correct
15 Correct 147 ms 209436 KB Output is correct
16 Correct 158 ms 209464 KB Output is correct
17 Correct 141 ms 209588 KB Output is correct
18 Correct 145 ms 209480 KB Output is correct
19 Correct 132 ms 209464 KB Output is correct
20 Correct 136 ms 209420 KB Output is correct
21 Correct 132 ms 209412 KB Output is correct
22 Correct 150 ms 209472 KB Output is correct
23 Correct 140 ms 209452 KB Output is correct
24 Correct 154 ms 209340 KB Output is correct
25 Correct 142 ms 209404 KB Output is correct
26 Correct 137 ms 209508 KB Output is correct
27 Correct 128 ms 209420 KB Output is correct
28 Correct 135 ms 209416 KB Output is correct
29 Correct 140 ms 209412 KB Output is correct
30 Correct 147 ms 209392 KB Output is correct
31 Correct 147 ms 209512 KB Output is correct
32 Correct 131 ms 209432 KB Output is correct
33 Correct 156 ms 209404 KB Output is correct
34 Correct 131 ms 209448 KB Output is correct
35 Correct 129 ms 209440 KB Output is correct
36 Correct 145 ms 209432 KB Output is correct
37 Correct 139 ms 209348 KB Output is correct
38 Correct 128 ms 209320 KB Output is correct
39 Correct 135 ms 209452 KB Output is correct
40 Correct 128 ms 209416 KB Output is correct
41 Correct 130 ms 209324 KB Output is correct
42 Correct 129 ms 209364 KB Output is correct
43 Correct 135 ms 209320 KB Output is correct
44 Correct 128 ms 209328 KB Output is correct
45 Correct 144 ms 209456 KB Output is correct
46 Correct 160 ms 210096 KB Output is correct
47 Correct 163 ms 210028 KB Output is correct
48 Correct 156 ms 210072 KB Output is correct
49 Correct 139 ms 210212 KB Output is correct
50 Correct 132 ms 210120 KB Output is correct
51 Correct 145 ms 210268 KB Output is correct
52 Correct 132 ms 210136 KB Output is correct
53 Correct 168 ms 210156 KB Output is correct
54 Correct 134 ms 210076 KB Output is correct
55 Correct 135 ms 210144 KB Output is correct
56 Correct 137 ms 210020 KB Output is correct
57 Correct 171 ms 210288 KB Output is correct
58 Correct 165 ms 210248 KB Output is correct
59 Correct 168 ms 210368 KB Output is correct
60 Correct 168 ms 210248 KB Output is correct
61 Correct 139 ms 210040 KB Output is correct
62 Correct 136 ms 209936 KB Output is correct
63 Correct 139 ms 209956 KB Output is correct
64 Correct 156 ms 209944 KB Output is correct
65 Correct 151 ms 209984 KB Output is correct
66 Correct 147 ms 209944 KB Output is correct
67 Correct 163 ms 209920 KB Output is correct
68 Correct 134 ms 210120 KB Output is correct
69 Correct 141 ms 210068 KB Output is correct
70 Correct 132 ms 209952 KB Output is correct
71 Correct 138 ms 209896 KB Output is correct
72 Correct 160 ms 210328 KB Output is correct
73 Correct 154 ms 210288 KB Output is correct
74 Correct 164 ms 209864 KB Output is correct
75 Correct 146 ms 209908 KB Output is correct
76 Correct 132 ms 209864 KB Output is correct
77 Correct 141 ms 209884 KB Output is correct
78 Correct 144 ms 209852 KB Output is correct
79 Correct 131 ms 209844 KB Output is correct
80 Correct 136 ms 209736 KB Output is correct
81 Correct 135 ms 209864 KB Output is correct
82 Correct 137 ms 209932 KB Output is correct
83 Correct 138 ms 209828 KB Output is correct
84 Correct 136 ms 209788 KB Output is correct
85 Correct 136 ms 209736 KB Output is correct
86 Correct 139 ms 209840 KB Output is correct

Subtask #3
84.0 / 84.0

# Verdict Execution time Memory Grader output
1 Correct 131 ms 209480 KB Output is correct
2 Correct 130 ms 209380 KB Output is correct
3 Correct 132 ms 209552 KB Output is correct
4 Correct 143 ms 209436 KB Output is correct
5 Correct 152 ms 209528 KB Output is correct
6 Correct 140 ms 209416 KB Output is correct
7 Correct 145 ms 209516 KB Output is correct
8 Correct 137 ms 209364 KB Output is correct
9 Correct 134 ms 209480 KB Output is correct
10 Correct 129 ms 209460 KB Output is correct
11 Correct 135 ms 209472 KB Output is correct
12 Correct 130 ms 209368 KB Output is correct
13 Correct 133 ms 209540 KB Output is correct
14 Correct 129 ms 209404 KB Output is correct
15 Correct 147 ms 209436 KB Output is correct
16 Correct 158 ms 209464 KB Output is correct
17 Correct 141 ms 209588 KB Output is correct
18 Correct 145 ms 209480 KB Output is correct
19 Correct 132 ms 209464 KB Output is correct
20 Correct 136 ms 209420 KB Output is correct
21 Correct 132 ms 209412 KB Output is correct
22 Correct 150 ms 209472 KB Output is correct
23 Correct 140 ms 209452 KB Output is correct
24 Correct 154 ms 209340 KB Output is correct
25 Correct 142 ms 209404 KB Output is correct
26 Correct 137 ms 209508 KB Output is correct
27 Correct 128 ms 209420 KB Output is correct
28 Correct 135 ms 209416 KB Output is correct
29 Correct 140 ms 209412 KB Output is correct
30 Correct 147 ms 209392 KB Output is correct
31 Correct 147 ms 209512 KB Output is correct
32 Correct 131 ms 209432 KB Output is correct
33 Correct 156 ms 209404 KB Output is correct
34 Correct 131 ms 209448 KB Output is correct
35 Correct 129 ms 209440 KB Output is correct
36 Correct 145 ms 209432 KB Output is correct
37 Correct 139 ms 209348 KB Output is correct
38 Correct 128 ms 209320 KB Output is correct
39 Correct 135 ms 209452 KB Output is correct
40 Correct 128 ms 209416 KB Output is correct
41 Correct 130 ms 209324 KB Output is correct
42 Correct 129 ms 209364 KB Output is correct
43 Correct 135 ms 209320 KB Output is correct
44 Correct 128 ms 209328 KB Output is correct
45 Correct 144 ms 209456 KB Output is correct
46 Correct 160 ms 210096 KB Output is correct
47 Correct 163 ms 210028 KB Output is correct
48 Correct 156 ms 210072 KB Output is correct
49 Correct 139 ms 210212 KB Output is correct
50 Correct 132 ms 210120 KB Output is correct
51 Correct 145 ms 210268 KB Output is correct
52 Correct 132 ms 210136 KB Output is correct
53 Correct 168 ms 210156 KB Output is correct
54 Correct 134 ms 210076 KB Output is correct
55 Correct 135 ms 210144 KB Output is correct
56 Correct 137 ms 210020 KB Output is correct
57 Correct 171 ms 210288 KB Output is correct
58 Correct 165 ms 210248 KB Output is correct
59 Correct 168 ms 210368 KB Output is correct
60 Correct 168 ms 210248 KB Output is correct
61 Correct 139 ms 210040 KB Output is correct
62 Correct 136 ms 209936 KB Output is correct
63 Correct 139 ms 209956 KB Output is correct
64 Correct 156 ms 209944 KB Output is correct
65 Correct 151 ms 209984 KB Output is correct
66 Correct 147 ms 209944 KB Output is correct
67 Correct 163 ms 209920 KB Output is correct
68 Correct 134 ms 210120 KB Output is correct
69 Correct 141 ms 210068 KB Output is correct
70 Correct 132 ms 209952 KB Output is correct
71 Correct 138 ms 209896 KB Output is correct
72 Correct 160 ms 210328 KB Output is correct
73 Correct 154 ms 210288 KB Output is correct
74 Correct 164 ms 209864 KB Output is correct
75 Correct 146 ms 209908 KB Output is correct
76 Correct 132 ms 209864 KB Output is correct
77 Correct 141 ms 209884 KB Output is correct
78 Correct 144 ms 209852 KB Output is correct
79 Correct 131 ms 209844 KB Output is correct
80 Correct 136 ms 209736 KB Output is correct
81 Correct 135 ms 209864 KB Output is correct
82 Correct 137 ms 209932 KB Output is correct
83 Correct 138 ms 209828 KB Output is correct
84 Correct 136 ms 209788 KB Output is correct
85 Correct 136 ms 209736 KB Output is correct
86 Correct 139 ms 209840 KB Output is correct
87 Correct 181 ms 211144 KB Output is correct
88 Correct 296 ms 214880 KB Output is correct
89 Correct 854 ms 227760 KB Output is correct
90 Correct 814 ms 227776 KB Output is correct
91 Correct 804 ms 227892 KB Output is correct
92 Correct 284 ms 231700 KB Output is correct
93 Correct 287 ms 231672 KB Output is correct
94 Correct 286 ms 231700 KB Output is correct
95 Correct 281 ms 229124 KB Output is correct
96 Correct 338 ms 229492 KB Output is correct
97 Correct 337 ms 229576 KB Output is correct
98 Correct 337 ms 229472 KB Output is correct
99 Correct 298 ms 226732 KB Output is correct
100 Correct 1109 ms 231528 KB Output is correct
101 Correct 1506 ms 231760 KB Output is correct
102 Correct 1507 ms 231632 KB Output is correct
103 Correct 1480 ms 231768 KB Output is correct
104 Correct 343 ms 226788 KB Output is correct
105 Correct 333 ms 226776 KB Output is correct
106 Correct 339 ms 226788 KB Output is correct
107 Correct 347 ms 225008 KB Output is correct
108 Correct 462 ms 225012 KB Output is correct
109 Correct 784 ms 225832 KB Output is correct
110 Correct 234 ms 229088 KB Output is correct
111 Correct 299 ms 229140 KB Output is correct
112 Correct 252 ms 226888 KB Output is correct
113 Correct 235 ms 224136 KB Output is correct
114 Correct 1097 ms 231496 KB Output is correct
115 Correct 827 ms 229072 KB Output is correct
116 Correct 293 ms 224144 KB Output is correct
117 Correct 330 ms 224448 KB Output is correct
118 Correct 339 ms 223840 KB Output is correct
119 Correct 370 ms 223584 KB Output is correct
120 Correct 249 ms 222000 KB Output is correct
121 Correct 249 ms 221496 KB Output is correct
122 Correct 266 ms 221364 KB Output is correct
123 Correct 369 ms 224568 KB Output is correct
124 Correct 399 ms 223856 KB Output is correct
125 Correct 448 ms 223536 KB Output is correct
126 Correct 284 ms 222184 KB Output is correct
127 Correct 296 ms 221564 KB Output is correct
128 Correct 294 ms 221168 KB Output is correct