oj.uz

Submission #575599

# Submission time Handle Problem Language Result Execution time Memory
575599 2022-06-11T06:33:34 Z jack715 Dreaming (IOI13_dreaming) C++14
32 / 100
 40 ms 10784 KB 
#include "dreaming.h"
#include <bits/stdc++.h>
#define ll long long
#define pb push_back
#define pp pop_back
#define mp make_pair
#define bb back
#define ff first
#define ss second

using namespace std;

int build(int now, int par, vector<int>& dist, vector<vector<pair<int, int> > >& path) {
    dist[now] = 0;
    for (auto next : path[now]) {
        if (next.ff == par) continue; 
        dist[now] = max(dist[now], build(next.ff, now, dist, path)+next.ss);
    }
    return dist[now];
}

pair<int, int> find(int now, int par, int up, vector<int>& dist, vector<vector<pair<int, int> > >& path) {
    vector<pair<int, pair<int, int> > > child;
    for (auto next : path[now]) {
        if (next.ff == par) continue; 
        child.pb({dist[next.ff]+next.ss, next});
    }
    if (child.size() == 0)
        return {up, up};
    sort(child.begin(), child.end());
    for (int i = 0; i < child.size()-1; i++) {
        up = max(up, child[i].ff);
    }
    if (dist[child.back().ss.ff] >= child.back().ss.ss + up)
        return find(child.back().ss.ff, now, up+child.back().ss.ss, dist, path);
    // cout << now << ' ' << up << ' ' << child.back().ff << ' ' << par << '\n';
    return {max(up, child.back().ff), child.back().ff+up};
}

int travelTime(int N, int M, int L, int A[], int B[], int T[]) {
    vector<int> dist(N, -1), centre;
    vector<vector<pair<int, int> > > path(N);
    for (int i = 0; i < M; i++) {
        path[A[i]].pb({B[i], T[i]});
        path[B[i]].pb({A[i], T[i]});
    }
    int ans = 0;
    for (int i = 0; i < N; i++) {
        if (dist[i] == -1) {
            build(i, -1, dist, path);
            pair<int, int> ret = find(i, -1, 0, dist, path); 
            centre.pb(ret.ff);
            ans = max(ans, ret.ss);
        }
    }

    sort(centre.begin(), centre.end());
    // for (int a : centre)
    //     cout << a << ' ';
    // cout << '\n';
    int p = centre.back();
    for (int i = 0; i < centre.size()-1; i++) {
        ans = max(ans, p+centre[i]+L);
        p = max(p, centre[i]+L);
    }
    return ans;
}

Compilation message

dreaming.cpp: In function 'std::pair<int, int> find(int, int, int, std::vector<int>&, std::vector<std::vector<std::pair<int, int> > >&)':
dreaming.cpp:31:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, std::pair<int, int> > >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   31 |     for (int i = 0; i < child.size()-1; i++) {
      |                     ~~^~~~~~~~~~~~~~~~
dreaming.cpp: In function 'int travelTime(int, int, int, int*, int*, int*)':
dreaming.cpp:62:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   62 |     for (int i = 0; i < centre.size()-1; i++) {
      |                     ~~^~~~~~~~~~~~~~~~~

Subtask #1
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 37 ms 10784 KB Output is correct
2 Correct 39 ms 10300 KB Output is correct
3 Correct 27 ms 9160 KB Output is correct
4 Correct 5 ms 1748 KB Output is correct
5 Correct 4 ms 1236 KB Output is correct
6 Correct 9 ms 2600 KB Output is correct
7 Correct 1 ms 340 KB Output is correct
8 Correct 18 ms 4836 KB Output is correct
9 Correct 24 ms 7380 KB Output is correct
10 Correct 0 ms 340 KB Output is correct
11 Correct 35 ms 8000 KB Output is correct
12 Correct 40 ms 9360 KB Output is correct
13 Correct 1 ms 340 KB Output is correct

Subtask #2
0 / 10.0

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

Subtask #3
0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 37 ms 10784 KB Output is correct
2 Correct 39 ms 10300 KB Output is correct
3 Correct 27 ms 9160 KB Output is correct
4 Correct 5 ms 1748 KB Output is correct
5 Correct 4 ms 1236 KB Output is correct
6 Correct 9 ms 2600 KB Output is correct
7 Correct 1 ms 340 KB Output is correct
8 Correct 18 ms 4836 KB Output is correct
9 Correct 24 ms 7380 KB Output is correct
10 Correct 0 ms 340 KB Output is correct
11 Correct 35 ms 8000 KB Output is correct
12 Correct 40 ms 9360 KB Output is correct
13 Correct 1 ms 340 KB Output is correct
14 Correct 1 ms 212 KB Output is correct
15 Correct 0 ms 212 KB Output is correct
16 Correct 1 ms 212 KB Output is correct
17 Incorrect 0 ms 212 KB Output isn't correct
18 Halted 0 ms 0 KB -

Subtask #4
18.0 / 18.0

# Verdict Execution time Memory Grader output
1 Correct 20 ms 5516 KB Output is correct
2 Correct 23 ms 5592 KB Output is correct
3 Correct 20 ms 5580 KB Output is correct
4 Correct 20 ms 5536 KB Output is correct
5 Correct 21 ms 5496 KB Output is correct
6 Correct 31 ms 6140 KB Output is correct
7 Correct 20 ms 5816 KB Output is correct
8 Correct 19 ms 5424 KB Output is correct
9 Correct 19 ms 5400 KB Output is correct
10 Correct 21 ms 5740 KB Output is correct
11 Correct 0 ms 212 KB Output is correct
12 Correct 5 ms 3664 KB Output is correct
13 Correct 5 ms 3664 KB Output is correct
14 Correct 5 ms 3664 KB Output is correct
15 Correct 4 ms 3664 KB Output is correct
16 Correct 5 ms 3664 KB Output is correct
17 Correct 5 ms 3664 KB Output is correct
18 Correct 5 ms 3664 KB Output is correct
19 Correct 4 ms 3664 KB Output is correct
20 Correct 0 ms 212 KB Output is correct
21 Correct 0 ms 212 KB Output is correct
22 Correct 1 ms 340 KB Output is correct
23 Correct 5 ms 3664 KB Output is correct

Subtask #5
0 / 12.0

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

Subtask #6
0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 37 ms 10784 KB Output is correct
2 Correct 39 ms 10300 KB Output is correct
3 Correct 27 ms 9160 KB Output is correct
4 Correct 5 ms 1748 KB Output is correct
5 Correct 4 ms 1236 KB Output is correct
6 Correct 9 ms 2600 KB Output is correct
7 Correct 1 ms 340 KB Output is correct
8 Correct 18 ms 4836 KB Output is correct
9 Correct 24 ms 7380 KB Output is correct
10 Correct 0 ms 340 KB Output is correct
11 Correct 35 ms 8000 KB Output is correct
12 Correct 40 ms 9360 KB Output is correct
13 Correct 1 ms 340 KB Output is correct
14 Correct 1 ms 212 KB Output is correct
15 Correct 0 ms 212 KB Output is correct
16 Correct 1 ms 212 KB Output is correct
17 Incorrect 0 ms 212 KB Output isn't correct
18 Halted 0 ms 0 KB -