oj.uz

답안 #600303

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
600303 2022-07-20T17:26:14 Z CaoHuuKhuongDuy Cities (BOI16_cities) C++17
74 / 100
 6000 ms 240228 KB 
#include <bits/stdc++.h>

#pragma GCC target("avx2")
#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")

using namespace std;

#define ii pair<int , int>  
#define iv pair<ii , ii>
#define iii pair<int , ii>
#define fi first
#define se second
#define int long long
const int inf = 1e18 + 7;
const int MAX_N = 1e5 + 7;
const int MOD = 1e9 + 7;

int n , m , k; 
vector<ii> adj[MAX_N];
int f[MAX_N][(1 << 5) + 1];
int a[MAX_N];
vector<int> mm[50];

signed main() {
    ios_base::sync_with_stdio(0); cin.tie(NULL); cout.tie(NULL);
    cin >> n >> k >> m;
    for (int i = 1 ; i <= n ;i ++) {
        for (int j = 1 ; j < (1 << k) ; j++) {
            f[i][j] = inf;
        }
    }
    priority_queue<iii , vector<iii> , greater<iii>> pq;
    for (int i = 0 ; i < k ; i++) {
        cin >> a[i];
        f[a[i]][1 << i] = 0;
        pq.push(iii(0 , ii(a[i] , 1 << i)));
    }
    for (int i = 1 ; i <= m ; i++) {
        int u , v , w; cin >> u >> v >> w;
        adj[u].push_back(ii(v , w));
        adj[v].push_back(ii(u , w));
    }
    for (int mask1 = 0 ; mask1 < (1 << k) ; mask1++) {
        for (int mask2 = 0 ; mask2 < (1 << k) ; mask2 ++) {
            if (!(mask1 & mask2)) {
                mm[mask1].push_back(mask2);
            }
        }
    }
    while (!pq.empty()) {
        int u = pq.top().se.fi;
        int mask1 = pq.top().se.se;
        int du = pq.top().fi;
        pq.pop();
        if (du != f[u][mask1]) continue;;
        for (ii edge : adj[u]) {
            int v , w; 
            tie(v , w) = edge;
            for (int mask2 : mm[mask1]) {
                if (!(mask1 & mask2) && (f[u][mask1] + f[v][mask2] + w) < f[v][mask1|mask2]) {
                    f[v][mask1|mask2] = f[u][mask1] + f[v][mask2] + w;
                    pq.push(iii(f[v][mask1|mask2] , ii(v , mask1|mask2)));
                }
            }
        }
    }
    int ans = inf;
    for (int i = 1 ; i <= n ; i++) {
        ans = min(ans , f[i][(1 << k) - 1]);
    }   
    cout << ans;
    return 0;
}

Subtask #1
22.0 / 22.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 2644 KB Output is correct
2 Correct 2 ms 2644 KB Output is correct
3 Correct 2 ms 2680 KB Output is correct
4 Correct 2 ms 2696 KB Output is correct
5 Correct 2 ms 2772 KB Output is correct

Subtask #2
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1018 ms 67788 KB Output is correct
2 Correct 848 ms 67204 KB Output is correct
3 Correct 436 ms 42144 KB Output is correct
4 Correct 109 ms 16732 KB Output is correct
5 Correct 380 ms 49260 KB Output is correct
6 Correct 87 ms 16036 KB Output is correct
7 Correct 5 ms 3340 KB Output is correct
8 Correct 4 ms 3156 KB Output is correct

Subtask #3
15.0 / 15.0

# 결과 실행 시간 메모리 Grader output
1 Correct 11 ms 3920 KB Output is correct
2 Correct 10 ms 3924 KB Output is correct
3 Correct 7 ms 3452 KB Output is correct
4 Correct 8 ms 3840 KB Output is correct

Subtask #4
23.0 / 23.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2818 ms 141696 KB Output is correct
2 Correct 2749 ms 140988 KB Output is correct
3 Correct 1384 ms 60600 KB Output is correct
4 Correct 1827 ms 79632 KB Output is correct
5 Correct 459 ms 42676 KB Output is correct
6 Correct 155 ms 20616 KB Output is correct

Subtask #5
0 / 26.0

# 결과 실행 시간 메모리 Grader output
1 Execution timed out 6068 ms 240228 KB Time limit exceeded
2 Halted 0 ms 0 KB -