oj.uz

Submission #1027441

# Submission time Handle Problem Language Result Execution time Memory
1027441 2024-07-19T06:27:43 Z onbert Race (IOI11_race) C++17
100 / 100
 1557 ms 225768 KB 
#include "race.h"
#include <bits/stdc++.h>
using namespace std;
#define int long long
struct thing {
    int dist, d, u;
    bool operator < (const thing &b) const {
        if (dist!=b.dist) return dist < b.dist;
        if (d!=b.d) return d < b.d;
        return u < b.u;
    }
};
struct node {
    pair<int,int> val;
    int lpt, rpt;
};

const int maxn = 2e5 + 5, maxN = 8e5 + 5, INF = 1e18;
int n, k;
vector<pair<int,int>> adj[maxn];
int sz[maxn], dist[maxn], d[maxn];
bool cmp(pair<int,int> x, pair<int,int> y) {
    return sz[x.first] > sz[y.first];
}

void dfs0(int u) {
    sz[u] = 1;
    for (auto [v, w]:adj[u]) {
        adj[v].erase(find(adj[v].begin(), adj[v].end(), make_pair(u, w)));
        dist[v] = dist[u] + w, d[v] = d[u] + 1;
        dfs0(v);
        sz[u] += sz[v];
    }
    sort(adj[u].begin(), adj[u].end(), cmp);
}
int newid[maxn], oldid[maxn], lim[maxn], cnt = -1;
void dfs1(int u) {
    cnt++;
    newid[u] = cnt, oldid[cnt] = u;
    for (auto [v, w]:adj[u]) dfs1(v);
    lim[u] = cnt;
}

vector<node> seg[maxN];
void build(int id, int l, int r) {
    seg[id] = {{{-INF, -INF}, 0, 0}};
    if (l==r) return;
    int mid = (l+r)/2;
    build(id*2, l, mid); build(id*2+1, mid+1, r);
}
void update(int id, int l, int r, int target, pair<int,int> val) {
    if (r<target || target<l) return;
    if (l==r) {
        seg[id].push_back({val, 0, 0});
        return;
    }
    int mid = (l+r)/2;
    update(id*2, l, mid, target, val); update(id*2+1, mid+1, r, target, val);
    pair<int,int> cur = max(seg[id*2].back().val, seg[id*2+1].back().val);
    seg[id].push_back({cur, (int)seg[id*2].size()-1, (int)seg[id*2+1].size()-1});
}
pair<int,int> qry(int id, int pt, int l, int r, int findl, int findr) {
    if (r<findl || findr<l) return {-INF, -INF};
    if (findl<=l && r<=findr) return seg[id][pt].val;
    int mid = (l+r)/2;
    pair<int,int> lhs = qry(id*2, seg[id][pt].lpt, l, mid, findl, findr);
    pair<int,int> rhs = qry(id*2+1, seg[id][pt].rpt, mid+1, r, findl, findr);
    return max(lhs, rhs);
}

int ans = INF, curD;
vector<thing> vec = {{-INF, -INF, -INF}};
void QRY(int l, int r, int val) {
    if (val<0) return;
    thing cur = {val, INF, INF};
    int pos = upper_bound(vec.begin(), vec.end(), cur) - vec.begin() - 1;
    // cout << l << " " << r << " " << val << " " << pos << endl;
    pair<int,int> temp = qry(1, pos, 0, n-1, l, r);
    // cout << temp.first << " " << temp.second << endl;
    if (temp.first!=val) return;
    ans = min(-temp.second + curD, ans);
}

int32_t best_path(int32_t N, int32_t K, int32_t H[][2], int32_t L[]) {
    n = N, k = K;
    for (int i=0;i<n-1;i++) {
        auto [u, v] = H[i];
        adj[u].push_back({v, L[i]});
        adj[v].push_back({u, L[i]});
    }
    dist[0] = 0, d[0] = 0;
    dfs0(0); dfs1(0);
    // for (int i=0;i<n;i++) cout << newid[i] << " "; cout << endl;
    for (int i=0;i<n;i++) vec.push_back({dist[i], -d[i], newid[i]});
    sort(vec.begin(), vec.end());
    build(1, 0, n-1);
    for (int i=1;i<=n;i++) {
        update(1, 0, n-1, vec[i].u, {vec[i].dist, vec[i].d});
        // cout << i << " " << vec[i].u << " " << vec[i].dist << " " << vec[i].d << endl;
    }
    for (int u=0;u<n;u++) {
        curD = -d[u];
        // if (u==6) cout << "TEST" << endl;
        QRY(newid[u]+1, lim[u], k + dist[u]);
        // if (u==6) cout << k+dist[u] << endl;
        for (int i=1;i<adj[u].size();i++) {
            int v = adj[u][i].first;
            for (int j=newid[v];j<=lim[v];j++) {
                curD = -2*d[u] + d[oldid[j]];
                QRY(newid[u]+1, newid[v]-1, k + 2*dist[u] - dist[oldid[j]]);
            }
        }
    }
    if (ans==INF) return -1;
    return ans;
}

Compilation message

race.cpp: In function 'int32_t best_path(int32_t, int32_t, int32_t (*)[2], int32_t*)':
race.cpp:106:23: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  106 |         for (int i=1;i<adj[u].size();i++) {
      |                      ~^~~~~~~~~~~~~~

Subtask #1
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 6 ms 36956 KB Output is correct
2 Correct 6 ms 37016 KB Output is correct
3 Correct 6 ms 36948 KB Output is correct
4 Correct 5 ms 32856 KB Output is correct
5 Correct 6 ms 32932 KB Output is correct
6 Correct 8 ms 32860 KB Output is correct
7 Correct 6 ms 34908 KB Output is correct
8 Correct 5 ms 32856 KB Output is correct
9 Correct 6 ms 32860 KB Output is correct
10 Correct 6 ms 32860 KB Output is correct
11 Correct 6 ms 36956 KB Output is correct
12 Correct 6 ms 34908 KB Output is correct
13 Correct 6 ms 34908 KB Output is correct
14 Correct 6 ms 34852 KB Output is correct
15 Correct 5 ms 32860 KB Output is correct
16 Correct 5 ms 32860 KB Output is correct
17 Correct 6 ms 32860 KB Output is correct
18 Correct 6 ms 34896 KB Output is correct

Subtask #2
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 6 ms 36956 KB Output is correct
2 Correct 6 ms 37016 KB Output is correct
3 Correct 6 ms 36948 KB Output is correct
4 Correct 5 ms 32856 KB Output is correct
5 Correct 6 ms 32932 KB Output is correct
6 Correct 8 ms 32860 KB Output is correct
7 Correct 6 ms 34908 KB Output is correct
8 Correct 5 ms 32856 KB Output is correct
9 Correct 6 ms 32860 KB Output is correct
10 Correct 6 ms 32860 KB Output is correct
11 Correct 6 ms 36956 KB Output is correct
12 Correct 6 ms 34908 KB Output is correct
13 Correct 6 ms 34908 KB Output is correct
14 Correct 6 ms 34852 KB Output is correct
15 Correct 5 ms 32860 KB Output is correct
16 Correct 5 ms 32860 KB Output is correct
17 Correct 6 ms 32860 KB Output is correct
18 Correct 6 ms 34896 KB Output is correct
19 Correct 5 ms 32860 KB Output is correct
20 Correct 6 ms 34860 KB Output is correct
21 Correct 7 ms 35420 KB Output is correct
22 Correct 7 ms 35420 KB Output is correct
23 Correct 7 ms 35564 KB Output is correct
24 Correct 8 ms 37464 KB Output is correct
25 Correct 7 ms 35420 KB Output is correct
26 Correct 7 ms 35416 KB Output is correct
27 Correct 8 ms 35416 KB Output is correct
28 Correct 8 ms 35420 KB Output is correct
29 Correct 8 ms 35420 KB Output is correct
30 Correct 7 ms 33372 KB Output is correct
31 Correct 6 ms 33372 KB Output is correct
32 Correct 7 ms 33372 KB Output is correct
33 Correct 7 ms 37452 KB Output is correct
34 Correct 7 ms 33376 KB Output is correct
35 Correct 7 ms 35420 KB Output is correct
36 Correct 7 ms 35420 KB Output is correct
37 Correct 7 ms 35416 KB Output is correct
38 Correct 8 ms 33372 KB Output is correct

Subtask #3
22.0 / 22.0

# Verdict Execution time Memory Grader output
1 Correct 6 ms 36956 KB Output is correct
2 Correct 6 ms 37016 KB Output is correct
3 Correct 6 ms 36948 KB Output is correct
4 Correct 5 ms 32856 KB Output is correct
5 Correct 6 ms 32932 KB Output is correct
6 Correct 8 ms 32860 KB Output is correct
7 Correct 6 ms 34908 KB Output is correct
8 Correct 5 ms 32856 KB Output is correct
9 Correct 6 ms 32860 KB Output is correct
10 Correct 6 ms 32860 KB Output is correct
11 Correct 6 ms 36956 KB Output is correct
12 Correct 6 ms 34908 KB Output is correct
13 Correct 6 ms 34908 KB Output is correct
14 Correct 6 ms 34852 KB Output is correct
15 Correct 5 ms 32860 KB Output is correct
16 Correct 5 ms 32860 KB Output is correct
17 Correct 6 ms 32860 KB Output is correct
18 Correct 6 ms 34896 KB Output is correct
19 Correct 405 ms 119224 KB Output is correct
20 Correct 407 ms 122536 KB Output is correct
21 Correct 435 ms 122552 KB Output is correct
22 Correct 449 ms 122448 KB Output is correct
23 Correct 719 ms 122804 KB Output is correct
24 Correct 576 ms 122524 KB Output is correct
25 Correct 299 ms 125612 KB Output is correct
26 Correct 148 ms 129208 KB Output is correct
27 Correct 541 ms 210552 KB Output is correct
28 Correct 907 ms 225448 KB Output is correct
29 Correct 912 ms 224676 KB Output is correct
30 Correct 549 ms 211280 KB Output is correct
31 Correct 536 ms 211120 KB Output is correct
32 Correct 925 ms 211168 KB Output is correct
33 Correct 1089 ms 210428 KB Output is correct
34 Correct 1293 ms 210212 KB Output is correct

Subtask #4
57.0 / 57.0

# Verdict Execution time Memory Grader output
1 Correct 6 ms 36956 KB Output is correct
2 Correct 6 ms 37016 KB Output is correct
3 Correct 6 ms 36948 KB Output is correct
4 Correct 5 ms 32856 KB Output is correct
5 Correct 6 ms 32932 KB Output is correct
6 Correct 8 ms 32860 KB Output is correct
7 Correct 6 ms 34908 KB Output is correct
8 Correct 5 ms 32856 KB Output is correct
9 Correct 6 ms 32860 KB Output is correct
10 Correct 6 ms 32860 KB Output is correct
11 Correct 6 ms 36956 KB Output is correct
12 Correct 6 ms 34908 KB Output is correct
13 Correct 6 ms 34908 KB Output is correct
14 Correct 6 ms 34852 KB Output is correct
15 Correct 5 ms 32860 KB Output is correct
16 Correct 5 ms 32860 KB Output is correct
17 Correct 6 ms 32860 KB Output is correct
18 Correct 6 ms 34896 KB Output is correct
19 Correct 5 ms 32860 KB Output is correct
20 Correct 6 ms 34860 KB Output is correct
21 Correct 7 ms 35420 KB Output is correct
22 Correct 7 ms 35420 KB Output is correct
23 Correct 7 ms 35564 KB Output is correct
24 Correct 8 ms 37464 KB Output is correct
25 Correct 7 ms 35420 KB Output is correct
26 Correct 7 ms 35416 KB Output is correct
27 Correct 8 ms 35416 KB Output is correct
28 Correct 8 ms 35420 KB Output is correct
29 Correct 8 ms 35420 KB Output is correct
30 Correct 7 ms 33372 KB Output is correct
31 Correct 6 ms 33372 KB Output is correct
32 Correct 7 ms 33372 KB Output is correct
33 Correct 7 ms 37452 KB Output is correct
34 Correct 7 ms 33376 KB Output is correct
35 Correct 7 ms 35420 KB Output is correct
36 Correct 7 ms 35420 KB Output is correct
37 Correct 7 ms 35416 KB Output is correct
38 Correct 8 ms 33372 KB Output is correct
39 Correct 405 ms 119224 KB Output is correct
40 Correct 407 ms 122536 KB Output is correct
41 Correct 435 ms 122552 KB Output is correct
42 Correct 449 ms 122448 KB Output is correct
43 Correct 719 ms 122804 KB Output is correct
44 Correct 576 ms 122524 KB Output is correct
45 Correct 299 ms 125612 KB Output is correct
46 Correct 148 ms 129208 KB Output is correct
47 Correct 541 ms 210552 KB Output is correct
48 Correct 907 ms 225448 KB Output is correct
49 Correct 912 ms 224676 KB Output is correct
50 Correct 549 ms 211280 KB Output is correct
51 Correct 536 ms 211120 KB Output is correct
52 Correct 925 ms 211168 KB Output is correct
53 Correct 1089 ms 210428 KB Output is correct
54 Correct 1293 ms 210212 KB Output is correct
55 Correct 33 ms 44688 KB Output is correct
56 Correct 28 ms 44880 KB Output is correct
57 Correct 375 ms 121932 KB Output is correct
58 Correct 305 ms 123740 KB Output is correct
59 Correct 158 ms 128948 KB Output is correct
60 Correct 505 ms 225768 KB Output is correct
61 Correct 572 ms 210340 KB Output is correct
62 Correct 506 ms 212356 KB Output is correct
63 Correct 659 ms 211120 KB Output is correct
64 Correct 882 ms 211368 KB Output is correct
65 Correct 1557 ms 213416 KB Output is correct
66 Correct 716 ms 222128 KB Output is correct
67 Correct 686 ms 215204 KB Output is correct
68 Correct 827 ms 211116 KB Output is correct
69 Correct 797 ms 208744 KB Output is correct
70 Correct 768 ms 206244 KB Output is correct