oj.uz

Submission #314995

# Submission time Handle Problem Language Result Execution time Memory
314995 2020-10-21T21:51:32 Z VROOM_VARUN Crocodile's Underground City (IOI11_crocodile) C++14
100 / 100
 906 ms 86120 KB 
/*
ID: varunra2
LANG: C++
TASK: crocodile
*/

#include <bits/stdc++.h>
using namespace std;

#ifdef DEBUG
#include "lib/debug.h"
#define debug(...) cerr << "[" << #__VA_ARGS__ << "]:", debug_out(__VA_ARGS__)
#define debug_arr(...) \
  cerr << "[" << #__VA_ARGS__ << "]:", debug_arr(__VA_ARGS__)
#pragma GCC diagnostic ignored "-Wsign-compare"
//#pragma GCC diagnostic ignored "-Wunused-parameter"
//#pragma GCC diagnostic ignored "-Wunused-variable"
#else
#define debug(...) 42
#endif

#define int long long

#define EPS 1e-9
#define IN(A, B, C) assert(B <= A && A <= C)
#define INF (int)1e9
#define MEM(a, b) memset(a, (b), sizeof(a))
#define MOD 1000000007
#define MP make_pair
#define PB push_back
#define all(cont) cont.begin(), cont.end()
#define rall(cont) cont.end(), cont.begin()
#define x first
#define y second

const double PI = acos(-1.0);
typedef long long ll;
typedef long double ld;
typedef pair<int, int> PII;
typedef map<int, int> MPII;
typedef multiset<int> MSETI;
typedef set<int> SETI;
typedef set<string> SETS;
typedef vector<int> VI;
typedef vector<PII> VII;
typedef vector<VI> VVI;
typedef vector<string> VS;

#define rep(i, a, b) for (int i = a; i < (b); ++i)
#define trav(a, x) for (auto& a : x)
#define sz(x) (int)(x).size()
typedef pair<int, int> pii;
typedef vector<int> vi;
#pragma GCC diagnostic ignored "-Wsign-compare"
// util functions

int32_t travel_plan(int32_t n, int32_t m, int32_t r[][2], int32_t l[], int32_t k, int32_t p[]) {
  vector<VII> adj(n);
  for (int i = 0; i < m; i++) {
    int u, v, w;
    u = r[i][0];
    v = r[i][1];
    w = l[i];
    adj[u].PB(MP(v, w));
    adj[v].PB(MP(u, w));
  }
  priority_queue<PII, VII, greater<PII>> pq;

  VI cnt(n, 2);

  VI dist(n, INF);

  for (int i = 0; i < k; i++) {
    pq.push(MP(0, p[i]));
    dist[p[i]] = 0;
  }

  vector<priority_queue<int>> dists(n);

  vector<bool> vis(n, false);

  while (!pq.empty()) {
    int u, w;
    u = pq.top().y;
    w = pq.top().x;
    pq.pop();
    if(vis[u]) continue;
    vis[u] = true;
    trav(x, adj[u]) {
      if (sz(dists[x.x]) < 2)
        dists[x.x].push(w + x.y);
      else if (dists[x.x].top() > w + x.y) {
        dists[x.x].pop();
        dists[x.x].push(w + x.y);
      }
      if (sz(dists[x.x]) >= 2 and dist[x.x] > dists[x.x].top()) {
        dist[x.x] = dists[x.x].top();
        pq.push(MP(dist[x.x], x.x));
      }
    }
  }

  return dists[0].top();
}

// int main() {
// #ifndef ONLINE_JUDGE
//   freopen("crocodile.in", "r", stdin);
//   freopen("crocodile.out", "w", stdout);
// #endif
//   cin.sync_with_stdio(0);
//   cin.tie(0);

//   int n, m, k;

//   cin >> n >> m >> k;

//   int r[m][2];
//   int l[m];
//   int p[k];

//   for (int i = 0; i < m; i++) {
//     cin >> r[i][0] >> r[i][1] >> l[i];
//   }

//   for (int i = 0; i < k; i++) {
//     cin >> p[i];
//   }

//   int x = travel_plan(n, m, r, l, k, p);

//   debug(x);

//   return 0;
// }

Subtask #1
46.0 / 46.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 384 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 512 KB Output is correct
5 Correct 1 ms 512 KB Output is correct
6 Correct 1 ms 512 KB Output is correct
7 Correct 2 ms 512 KB Output is correct
8 Correct 1 ms 512 KB Output is correct

Subtask #2
43.0 / 43.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 384 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 512 KB Output is correct
5 Correct 1 ms 512 KB Output is correct
6 Correct 1 ms 512 KB Output is correct
7 Correct 2 ms 512 KB Output is correct
8 Correct 1 ms 512 KB Output is correct
9 Correct 3 ms 768 KB Output is correct
10 Correct 1 ms 384 KB Output is correct
11 Correct 1 ms 512 KB Output is correct
12 Correct 4 ms 1024 KB Output is correct
13 Correct 4 ms 1152 KB Output is correct
14 Correct 1 ms 384 KB Output is correct
15 Correct 1 ms 512 KB Output is correct

Subtask #3
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 384 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 512 KB Output is correct
5 Correct 1 ms 512 KB Output is correct
6 Correct 1 ms 512 KB Output is correct
7 Correct 2 ms 512 KB Output is correct
8 Correct 1 ms 512 KB Output is correct
9 Correct 3 ms 768 KB Output is correct
10 Correct 1 ms 384 KB Output is correct
11 Correct 1 ms 512 KB Output is correct
12 Correct 4 ms 1024 KB Output is correct
13 Correct 4 ms 1152 KB Output is correct
14 Correct 1 ms 384 KB Output is correct
15 Correct 1 ms 512 KB Output is correct
16 Correct 673 ms 76412 KB Output is correct
17 Correct 110 ms 23928 KB Output is correct
18 Correct 163 ms 25592 KB Output is correct
19 Correct 906 ms 86120 KB Output is correct
20 Correct 343 ms 65704 KB Output is correct
21 Correct 51 ms 9976 KB Output is correct
22 Correct 385 ms 63864 KB Output is correct