oj.uz

Submission #367699

# Submission time Handle Problem Language Result Execution time Memory
367699 2021-02-18T03:46:15 Z arujbansal Relay Marathon (NOI20_relaymarathon) C++17
5 / 100
 6000 ms 66688 KB 
#include <iostream>
#include <algorithm>
#include <vector>
#include <map>
#include <set>
#include <array>
#include <stack>
#include <queue>
#include <random>
#include <numeric>
#include <functional>
#include <chrono>
#include <utility>
#include <iomanip>
#include <assert.h>

using namespace std;

void dbg_out() { cerr << endl; }
template<typename Head, typename... Tail>
void dbg_out(Head H, Tail... T) { cerr << ' ' << H; dbg_out(T...); }
#define dbg(...) cerr << "(" << #__VA_ARGS__ << "):", dbg_out(__VA_ARGS__)

#define rng_init mt19937 rng(chrono::steady_clock::now().time_since_epoch().count())
#define rng_seed(x) mt19937 rng(x)
#define all(x) (x).begin(), (x).end()
#define sz(x) (int) (x).size()
// #define int long long

rng_init;

const int MXN = 1e5 + 5, INF = 1e9 + 5;
int N, M, K;
vector<pair<int, int>> g[MXN];
vector<int> src[2], sink[2];
int dist[2][MXN];

void dijkstra(int which) {
    priority_queue<pair<int, int>, vector<pair<int, int>>, greater<>> pq;

    for (int i = 0; i < N; i++)
        dist[which][i] = INF;
    
    for (const auto &x : src[which]) {
        dist[which][x] = 0;
        pq.emplace(0, x);
    }

    while (!pq.empty()) {
        auto cur = pq.top();
        pq.pop();

        int u = cur.second, cur_dist = cur.first;
        if (cur_dist > dist[which][u]) continue;

        for (const auto &[v, wt] : g[u]) {
            int new_dist = cur_dist + wt;

            if (new_dist < dist[which][v]) {
                dist[which][v] = new_dist;
                pq.emplace(new_dist, v);
            }
        }
    }
}

void solve() {
    cin >> N >> M >> K;

    for (int i = 0; i < M; i++) {
        int u, v, w;
        cin >> u >> v >> w;
        u--, v--;

        g[u].emplace_back(v, w);
        g[v].emplace_back(u, w);
    }

    vector<int> special(K);
    for (auto &x : special) {
        cin >> x;
        x--;
    }

    int ans = INF;

    for (int iter = 0; iter < 45; iter++) {
        for (int i = 0; i < 2; i++)
            src[i].clear(), sink[i].clear();

        shuffle(all(special), rng);
        shuffle(all(special), rng);
        shuffle(all(special), rng);

        for (int i = 0; i < K / 2; i++)
            src[i % 2].push_back(special[i]);

        for (int i = K / 2; i < K; i++)
            sink[i % 2].push_back(special[i]);

        for (int i = 0; i < 2; i++)
            dijkstra(i);

        int dist1 = INF, dist2 = INF;

        for (const auto &x : sink[0])
            dist1 = min(dist1, dist[0][x]);

        for (const auto &x : sink[1])
            dist2 = min(dist2, dist[1][x]);

        ans = min(ans, dist1 + dist2);
    }

    cout << ans;
}

signed main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    int TC = 1;
    // cin >> TC;
    while (TC--) solve();
}

Subtask #1
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 2668 KB Output is correct
2 Correct 4 ms 2796 KB Output is correct
3 Correct 3 ms 2668 KB Output is correct
4 Correct 2 ms 2668 KB Output is correct
5 Correct 2 ms 2668 KB Output is correct
6 Correct 3 ms 2668 KB Output is correct
7 Correct 4 ms 2796 KB Output is correct
8 Correct 4 ms 2796 KB Output is correct
9 Correct 2 ms 2668 KB Output is correct
10 Correct 2 ms 2668 KB Output is correct
11 Correct 3 ms 2668 KB Output is correct
12 Correct 3 ms 2668 KB Output is correct
13 Correct 4 ms 2796 KB Output is correct
14 Correct 4 ms 2816 KB Output is correct
15 Correct 4 ms 2796 KB Output is correct
16 Correct 3 ms 2668 KB Output is correct
17 Correct 4 ms 2668 KB Output is correct
18 Correct 2 ms 2668 KB Output is correct
19 Correct 5 ms 2796 KB Output is correct
20 Correct 4 ms 2668 KB Output is correct
21 Correct 4 ms 2668 KB Output is correct
22 Correct 2 ms 2668 KB Output is correct
23 Correct 3 ms 2668 KB Output is correct
24 Correct 3 ms 2668 KB Output is correct
25 Correct 2 ms 2668 KB Output is correct
26 Correct 2 ms 2668 KB Output is correct
27 Correct 2 ms 2668 KB Output is correct
28 Correct 4 ms 2796 KB Output is correct
29 Correct 4 ms 2796 KB Output is correct
30 Correct 4 ms 2816 KB Output is correct

Subtask #2
0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 2668 KB Output is correct
2 Correct 4 ms 2796 KB Output is correct
3 Correct 3 ms 2668 KB Output is correct
4 Correct 2 ms 2668 KB Output is correct
5 Correct 2 ms 2668 KB Output is correct
6 Correct 3 ms 2668 KB Output is correct
7 Correct 4 ms 2796 KB Output is correct
8 Correct 4 ms 2796 KB Output is correct
9 Correct 2 ms 2668 KB Output is correct
10 Correct 2 ms 2668 KB Output is correct
11 Correct 3 ms 2668 KB Output is correct
12 Correct 3 ms 2668 KB Output is correct
13 Correct 4 ms 2796 KB Output is correct
14 Correct 4 ms 2816 KB Output is correct
15 Correct 4 ms 2796 KB Output is correct
16 Correct 3 ms 2668 KB Output is correct
17 Correct 4 ms 2668 KB Output is correct
18 Correct 2 ms 2668 KB Output is correct
19 Correct 5 ms 2796 KB Output is correct
20 Correct 4 ms 2668 KB Output is correct
21 Correct 4 ms 2668 KB Output is correct
22 Correct 2 ms 2668 KB Output is correct
23 Correct 3 ms 2668 KB Output is correct
24 Correct 3 ms 2668 KB Output is correct
25 Correct 2 ms 2668 KB Output is correct
26 Correct 2 ms 2668 KB Output is correct
27 Correct 2 ms 2668 KB Output is correct
28 Correct 4 ms 2796 KB Output is correct
29 Correct 4 ms 2796 KB Output is correct
30 Correct 4 ms 2816 KB Output is correct
31 Correct 4 ms 2688 KB Output is correct
32 Correct 7 ms 2668 KB Output is correct
33 Correct 7 ms 2668 KB Output is correct
34 Correct 3 ms 2668 KB Output is correct
35 Correct 2 ms 2668 KB Output is correct
36 Correct 19 ms 2924 KB Output is correct
37 Correct 22 ms 2924 KB Output is correct
38 Correct 9 ms 2796 KB Output is correct
39 Correct 74 ms 4972 KB Output is correct
40 Correct 33 ms 3308 KB Output is correct
41 Correct 9 ms 2796 KB Output is correct
42 Correct 33 ms 3308 KB Output is correct
43 Incorrect 17 ms 2796 KB Output isn't correct
44 Halted 0 ms 0 KB -

Subtask #3
0 / 25.0

# Verdict Execution time Memory Grader output
1 Correct 1354 ms 6380 KB Output is correct
2 Correct 11 ms 3948 KB Output is correct
3 Execution timed out 6040 ms 66688 KB Time limit exceeded
4 Halted 0 ms 0 KB -

Subtask #4
0 / 58.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 2668 KB Output is correct
2 Correct 4 ms 2796 KB Output is correct
3 Correct 3 ms 2668 KB Output is correct
4 Correct 2 ms 2668 KB Output is correct
5 Correct 2 ms 2668 KB Output is correct
6 Correct 3 ms 2668 KB Output is correct
7 Correct 4 ms 2796 KB Output is correct
8 Correct 4 ms 2796 KB Output is correct
9 Correct 2 ms 2668 KB Output is correct
10 Correct 2 ms 2668 KB Output is correct
11 Correct 3 ms 2668 KB Output is correct
12 Correct 3 ms 2668 KB Output is correct
13 Correct 4 ms 2796 KB Output is correct
14 Correct 4 ms 2816 KB Output is correct
15 Correct 4 ms 2796 KB Output is correct
16 Correct 3 ms 2668 KB Output is correct
17 Correct 4 ms 2668 KB Output is correct
18 Correct 2 ms 2668 KB Output is correct
19 Correct 5 ms 2796 KB Output is correct
20 Correct 4 ms 2668 KB Output is correct
21 Correct 4 ms 2668 KB Output is correct
22 Correct 2 ms 2668 KB Output is correct
23 Correct 3 ms 2668 KB Output is correct
24 Correct 3 ms 2668 KB Output is correct
25 Correct 2 ms 2668 KB Output is correct
26 Correct 2 ms 2668 KB Output is correct
27 Correct 2 ms 2668 KB Output is correct
28 Correct 4 ms 2796 KB Output is correct
29 Correct 4 ms 2796 KB Output is correct
30 Correct 4 ms 2816 KB Output is correct
31 Correct 4 ms 2688 KB Output is correct
32 Correct 7 ms 2668 KB Output is correct
33 Correct 7 ms 2668 KB Output is correct
34 Correct 3 ms 2668 KB Output is correct
35 Correct 2 ms 2668 KB Output is correct
36 Correct 19 ms 2924 KB Output is correct
37 Correct 22 ms 2924 KB Output is correct
38 Correct 9 ms 2796 KB Output is correct
39 Correct 74 ms 4972 KB Output is correct
40 Correct 33 ms 3308 KB Output is correct
41 Correct 9 ms 2796 KB Output is correct
42 Correct 33 ms 3308 KB Output is correct
43 Incorrect 17 ms 2796 KB Output isn't correct
44 Halted 0 ms 0 KB -