oj.uz

답안 #485806

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
485806 2021-11-09T11:58:21 Z radal Cities (BOI16_cities) C++14
15 / 100
 319 ms 16672 KB 
#include <bits/stdc++.h>
#pragma GCC optimize("O2")
#pragma GCC target("avx2,fma")
#define rep(i,l,r) for (int i = l; i < r; i++)
#define repr(i,r,l) for (int i = r; i >= l; i--)
#define X first
#define Y second
#define mp make_pair
#define pb push_back
#define endl '\n'
#define debug(x) cerr << #x << " : " << x << endl;
using namespace std;
typedef long long ll;
typedef pair<ll,int> pll;
typedef pair<long double,long double> pld;
const long long int N = 1e3+10,mod = 998244353,inf = 1e18,maxm = (1 << 24);
inline int mkay(int a,int b){
    if (a+b >= mod) return a+b-mod;
    if (a+b < 0) return a+b+mod;
    return a+b;
}
inline int poww(int n,ll k){
    int c = 1;
    while (k){
        if (k&1) c = (1ll*c*n)%mod;
        n = (1ll*n*n)%mod;
        k >>= 1;
    }
    return c;
}
vector<pll> adj[N];
int n,k;
ll d[N][N];
void dij(int v,int i){
    d[v][i] = 0;
    priority_queue<pll> pq;
    pq.push({0,v});
    while (!pq.empty()){
        pll u = pq.top();
        pq.pop();
        if (-u.X != d[u.Y][i]) continue;
        for (pll v : adj[u.Y]){
            if (d[v.X][i] > v.Y+d[u.Y][i]){
                d[v.X][i] = v.Y+d[u.Y][i];
                pq.push({-d[v.X][i],v.X});
            }
        }
    }
}
int main(){
    ios :: sync_with_stdio(0); cin.tie(0); cout.tie(0);
    memset(d,63,sizeof d);
    int m;
    cin >> n >> k >> m;
    vector<int> ve;
    rep(i,0,k){
        int x;
        cin >> x;
        ve.pb(x);
    }
    rep(i,0,m){
        int u,v,w;
        cin >> u >> v >> w;
        adj[u].pb({v,w});
        adj[v].pb({u,w});
    }
    rep(i,1,n+1) dij(i,i);
    ll ans = inf;
    rep(i,1,n+1){
        ll cost = 0;
        rep(j,0,k) cost += d[i][j];
        ans = min(ans,cost);
    }
    if (k < 4){
        cout << ans;
        return 0;
    }
    sort(ve.begin(),ve.end());
    while (true) {
        ll cost = 0;
        rep(i,1,k) cost += d[ve[i]][ve[i-1]];
        ans = min(ans,cost);
        if (!next_permutation(ve.begin(),ve.end())) break;
    }
    rep(i,1,n+1){
        rep(j,0,k){
            ll cost = 0,mi = inf;
            rep(x,0,k){
                if (x == j) continue;
                cost += d[i][ve[x]];
                mi = min(mi,d[ve[j]][ve[x]]);
            }
            ans = min(ans,cost+mi);
        }
        rep(j,1,n+1){
            while (true){
                ans = min(ans,d[i][j]+d[i][ve[0]]+d[i][ve[1]]+d[j][ve[2]]+d[j][ve[3]]);
                if (!next_permutation(ve.begin(),ve.end())) break;
            }
        }
    }
    cout << ans;
    return 0;
}

Subtask #1
0 / 22.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 3 ms 8268 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 10 ms 16600 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #3
15.0 / 15.0

# 결과 실행 시간 메모리 Grader output
1 Correct 316 ms 8516 KB Output is correct
2 Correct 319 ms 8480 KB Output is correct
3 Correct 215 ms 8396 KB Output is correct
4 Correct 94 ms 8480 KB Output is correct

Subtask #4
0 / 23.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 13 ms 16648 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #5
0 / 26.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 10 ms 16672 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -