oj.uz

답안 #285337

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
285337 2020-08-28T18:12:04 Z thecodingwizard Cities (BOI16_cities) C++11
100 / 100
 5820 ms 103808 KB 
#include <bits/stdc++.h>
using namespace std;

#define ii pair<int, int>
#define f first
#define s second
#define inf 100000000000010
#define ll long long
#define li pair<ll, int>

int main() {
    cin.tie(0); ios_base::sync_with_stdio(0);
    int n, k, m; cin >> n >> k >> m;
    vector<int> A(k); for (int &x : A) { cin >> x; --x; }
    vector<vector<ii>> adj(n);
    for (int i = 0; i < m; i++) {
        int a, b, w; cin >> a >> b >> w; --a; --b;
        adj[a].push_back(make_pair(b, w));
        adj[b].push_back(make_pair(a, w));
    }
    ll cost[k][n];
    for (int i = 0; i < k; i++) {
        for (int j = 0; j < n; j++) cost[i][j] = inf;
        cost[i][A[i]] = 0;
        priority_queue<li, vector<li>, greater<li>> pq; pq.push(make_pair(0, A[i]));
        while (!pq.empty()) {
            li u = pq.top(); pq.pop();
            if (cost[i][u.s] < u.f) continue;
            for (ii v : adj[u.s]) {
                if (cost[i][v.f] > u.f + v.s) {
                    cost[i][v.f] = u.f + v.s;
                    pq.push(make_pair(u.f+v.s, v.f));
                }
            }
        }
    }
    ll dist[n][(1 << k)];
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < (1 << k); j++) {
            dist[i][j] = inf;
        }
    }
    priority_queue<pair<ll, ii>, vector<pair<ll, ii>>, greater<pair<ll, ii>>> pq;
    for (int i = 0; i < k; i++) {
        dist[A[i]][(1 << i)] = 0;
        pq.push(make_pair(0, make_pair(A[i], (1 << i))));
    }
    while (!pq.empty()) {
        pair<ll, ii> top = pq.top(); pq.pop();
        ii u = top.s;
        if (dist[u.f][u.s] < top.f) continue;

        for (ii v : adj[u.f]) {
            ii nxt = make_pair(v.f, u.s);
            ll d = dist[u.f][u.s] + v.s;
            if (dist[nxt.f][nxt.s] > d) {
                dist[nxt.f][nxt.s] = d;
                pq.push(make_pair(d, nxt));
            }
        }

        for (int i = 0; i < k; i++) {
            ii nxt = make_pair(u.f, u.s|(1 << i));
            ll d = dist[u.f][u.s] + cost[i][u.f];
            if (dist[nxt.f][nxt.s] > d) {
                dist[nxt.f][nxt.s] = d;
                pq.push(make_pair(d, nxt));
            }
        }
    }
    ll best = inf;
    int msk = (1 << k) - 1;
    for (int i = 0; i < n; i++) {
        best = min(best, dist[i][msk]);
    }
    cout << best << endl;

    return 0;
}

Subtask #1
22.0 / 22.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 416 KB Output is correct
2 Correct 0 ms 384 KB Output is correct
3 Correct 1 ms 384 KB Output is correct
4 Correct 1 ms 384 KB Output is correct
5 Correct 1 ms 384 KB Output is correct

Subtask #2
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1093 ms 25912 KB Output is correct
2 Correct 1095 ms 25372 KB Output is correct
3 Correct 623 ms 18664 KB Output is correct
4 Correct 87 ms 5412 KB Output is correct
5 Correct 479 ms 17828 KB Output is correct
6 Correct 79 ms 5348 KB Output is correct
7 Correct 5 ms 768 KB Output is correct
8 Correct 2 ms 512 KB Output is correct

Subtask #3
15.0 / 15.0

# 결과 실행 시간 메모리 Grader output
1 Correct 8 ms 1024 KB Output is correct
2 Correct 8 ms 1024 KB Output is correct
3 Correct 6 ms 896 KB Output is correct
4 Correct 5 ms 768 KB Output is correct

Subtask #4
23.0 / 23.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2553 ms 57504 KB Output is correct
2 Correct 2222 ms 57180 KB Output is correct
3 Correct 1812 ms 38100 KB Output is correct
4 Correct 1440 ms 31792 KB Output is correct
5 Correct 288 ms 15232 KB Output is correct
6 Correct 91 ms 7276 KB Output is correct

Subtask #5
26.0 / 26.0

# 결과 실행 시간 메모리 Grader output
1 Correct 5781 ms 103808 KB Output is correct
2 Correct 5820 ms 103808 KB Output is correct
3 Correct 5232 ms 103416 KB Output is correct
4 Correct 3864 ms 100760 KB Output is correct
5 Correct 3040 ms 55024 KB Output is correct
6 Correct 536 ms 16536 KB Output is correct
7 Correct 110 ms 7792 KB Output is correct