oj.uz

Submission #887279

# Submission time Handle Problem Language Result Execution time Memory
887279 2023-12-14T07:32:02 Z vjudge1 Cities (BOI16_cities) C++17
100 / 100
 2060 ms 52152 KB 
/// Seeing the beauty through the...
/// Pain!
#include <bits/stdc++.h>
//#include <ext/pb_ds/assoc_container.hpp>
//#include <ext/pb_ds/tree_policy.hpp>
#define F first
#define S second
#define pb push_back
#define ppb pop_back
#define fast_io ios::sync_with_stdio(false);cin.tie(NULL);
#define file_io freopen("input.txt","r",stdin);freopen("output.txt","w",stdout);
#define FOR(i,k,n) for(int i = k; i < n; ++ i)
#define debf cout<<"(0-0)\n";
#define all(x) x.begin(), x.end()
#define dec(x) cout << fixed << setprecision(x);
#define pf push_front
#define ppf pop_front
#define dash " ------- "
#define what(x) cerr << #x << " is " << x << endl;
#define eb emplace_back
//#define int short int
#define int long long
#define sz(s) (int) (s.size())
#define fl cout.flush()

using namespace std;
//using namespace __gnu_pbds;

typedef long long ll;
typedef pair <int, int> pii;
typedef pair <int, pii> pip;
typedef pair <pii, int> ppi;
typedef pair <ll, ll> pll;
typedef unsigned long long ull;
typedef long double ld;

template <class T> using max_heap = priority_queue <T, vector <T>, less <T> >;
template <class T> using min_heap = priority_queue <T, vector <T>, greater <T> >;
//template <class T> using ordered_set = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;

constexpr int MOD = 1e9 + 7, N = 2e5 + 8, M = 1e5 + 4, SQ = 300, INF = 1e15 + 8, LGN = 22, mod = 998244353, P = 131113;

mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());

bool mark[N];
int n, k, m, dist[N][(1 << 5) + 1], ans = INF;
vector <pii> adj[N];
min_heap <pii> pq;

void dijk (int mask){
  for (int i = 0; i < n; ++ i){
    mark[i] = false;
    if (dist[i][mask] < INF){
      pq.push({dist[i][mask], i});
    }
  }
  while (pq.size()){
    pii p = pq.top();
    pq.pop();
    int x = p.S;
    if (mark[x]){
      continue;
    }
    mark[x] = true;
    for (pii i : adj[x]){
      if (dist[i.S][mask] > dist[x][mask] + i.F){
        dist[i.S][mask] = dist[x][mask] + i.F;
        pq.push({dist[i.S][mask], i.S});
      }
    }
  }
  return;
}

int32_t main(){
  fast_io;
  cin >> n >> k >> m;
  FOR (i, 0, n){
    FOR (j, 0, (1 << k)){
      dist[i][j] = INF;
    }
  }
  //int tt = 0;
  FOR (i, 0, k){
    int aa;
    //tt = aa - 1;
    cin >> aa;
    dist[--aa][(1<<i)] = 0;
  }
  FOR (i, 0, m){
    int aa, bb, cc;
    cin >> aa >> bb >> cc;
    adj[--aa].pb({cc, -- bb});
    adj[bb].pb({cc, aa});
  }
  for (int mask = 0; mask < (1 << k); ++ mask){
    for (int u = 0; u < n; ++ u){
      for (int sub = mask; sub > 0; sub = (sub - 1) & mask){
        dist[u][mask] = min (dist[u][mask], dist[u][sub] + dist[u][mask^sub]);
      }
    }
    dijk (mask);
  }
  /*for (int i = 0; i < n; ++ i){
    for (int j = 0; j < (1 << k); ++ j){
      cout << dist[i][j] << " ";
    }
    cout << '\n';
  }*/
  for (int i = 0; i < n; ++ i){
    ans = min (ans, dist[i][(1<<k)-1]);
  }
  cout << ans;
  return 0;
}

// Yesterday is history
// Tomorrow is a mystery
// but today is a gift
// That is why it is called the present

Subtask #1
22.0 / 22.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 6744 KB Output is correct
2 Correct 1 ms 6744 KB Output is correct
3 Correct 1 ms 6748 KB Output is correct
4 Correct 1 ms 6748 KB Output is correct
5 Correct 1 ms 6748 KB Output is correct

Subtask #2
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 505 ms 46792 KB Output is correct
2 Correct 453 ms 50924 KB Output is correct
3 Correct 264 ms 40900 KB Output is correct
4 Correct 47 ms 19212 KB Output is correct
5 Correct 243 ms 51520 KB Output is correct
6 Correct 45 ms 19024 KB Output is correct
7 Correct 3 ms 7000 KB Output is correct
8 Correct 2 ms 7004 KB Output is correct

Subtask #3
15.0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 5 ms 6744 KB Output is correct
2 Correct 5 ms 7004 KB Output is correct
3 Correct 4 ms 6748 KB Output is correct
4 Correct 4 ms 6744 KB Output is correct

Subtask #4
23.0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 979 ms 46400 KB Output is correct
2 Correct 952 ms 50148 KB Output is correct
3 Correct 628 ms 41180 KB Output is correct
4 Correct 515 ms 35968 KB Output is correct
5 Correct 143 ms 22256 KB Output is correct
6 Correct 52 ms 21328 KB Output is correct

Subtask #5
26.0 / 26.0

# Verdict Execution time Memory Grader output
1 Correct 2048 ms 47428 KB Output is correct
2 Correct 2060 ms 52152 KB Output is correct
3 Correct 1974 ms 50944 KB Output is correct
4 Correct 1342 ms 40840 KB Output is correct
5 Correct 982 ms 35844 KB Output is correct
6 Correct 198 ms 22612 KB Output is correct
7 Correct 61 ms 21584 KB Output is correct